Spaces:
Runtime error
Runtime error
change model
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ model = None
|
|
19 |
tokenizer = None
|
20 |
generator = None
|
21 |
|
22 |
-
os.environ["CUDA_VISIBLE_DEVICES"]="
|
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("
|
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>
|