Spaces:
Runtime error
Runtime error
fixed model
Browse files
app.py
CHANGED
@@ -10,12 +10,13 @@ llm = Llama(model_path="./ggjt-model.bin")
|
|
10 |
|
11 |
fixed_instruction = "You are a healthcare bot designed to give advice for the prevention and treatment of various illnesses."
|
12 |
|
13 |
-
def respond(message):
|
14 |
full_instruction = fixed_instruction + " " + message
|
15 |
bot_message = llm(full_instruction, stop=['### Instruction:', '### End'])
|
16 |
bot_message = bot_message['choices'][0]['text']
|
17 |
return bot_message
|
18 |
|
|
|
19 |
gr.ChatInterface(
|
20 |
fn=respond,
|
21 |
chatbot=gr.Chatbot(height=300),
|
|
|
10 |
|
11 |
fixed_instruction = "You are a healthcare bot designed to give advice for the prevention and treatment of various illnesses."
|
12 |
|
13 |
+
def respond(message, chat_history):
|
14 |
full_instruction = fixed_instruction + " " + message
|
15 |
bot_message = llm(full_instruction, stop=['### Instruction:', '### End'])
|
16 |
bot_message = bot_message['choices'][0]['text']
|
17 |
return bot_message
|
18 |
|
19 |
+
|
20 |
gr.ChatInterface(
|
21 |
fn=respond,
|
22 |
chatbot=gr.Chatbot(height=300),
|