sjdata commited on
Commit
20c5a04
1 Parent(s): 22fe0e9

fixed model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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),