Robert Pardela commited on
Commit
03640b4
·
1 Parent(s): 7408af2
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def vanilla_chatbot(message, history):
14
  return conversation.generated_responses[-1]
15
 
16
  def chat_bot(message, history):
17
- return chatbot(message, min_length=5, max_length=5000)[0]
18
 
19
  demo_chatbot = gr.ChatInterface(chat_bot, title="Check medical chatbot", description="Enter question")
20
 
 
14
  return conversation.generated_responses[-1]
15
 
16
  def chat_bot(message, history):
17
+ return chatbot(message, max_length=500)[0]['generated_text']
18
 
19
  demo_chatbot = gr.ChatInterface(chat_bot, title="Check medical chatbot", description="Enter question")
20