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),
|