ttagu99 commited on
Commit
3e95aa0
β€’
1 Parent(s): b2a459a

change model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,7 +19,7 @@ model = None
19
  tokenizer = None
20
  generator = None
21
 
22
- os.environ["CUDA_VISIBLE_DEVICES"]="1"
23
 
24
  def load_model(model_name, eight_bit=0, device_map="auto"):
25
  global model, tokenizer, generator
@@ -104,7 +104,7 @@ def predict(input, chatbot, state):
104
  chatbot.append((input, ko_response))
105
  return chatbot, state
106
 
107
- load_model("zl111/ChatDoctor")
108
 
109
  with gr.Blocks() as demo:
110
  gr.Markdown("""<h1><center>μ±— λ‹₯ν„°μž…λ‹ˆλ‹€. μ–΄λ””κ°€ λΆˆνŽΈν•˜μ‹ κ°€μš”?</center></h1>
 
19
  tokenizer = None
20
  generator = None
21
 
22
+ os.environ["CUDA_VISIBLE_DEVICES"]="0"
23
 
24
  def load_model(model_name, eight_bit=0, device_map="auto"):
25
  global model, tokenizer, generator
 
104
  chatbot.append((input, ko_response))
105
  return chatbot, state
106
 
107
+ load_model("mnc-ai/chatdoctor")
108
 
109
  with gr.Blocks() as demo:
110
  gr.Markdown("""<h1><center>μ±— λ‹₯ν„°μž…λ‹ˆλ‹€. μ–΄λ””κ°€ λΆˆνŽΈν•˜μ‹ κ°€μš”?</center></h1>