anezatra commited on
Commit
a29bf44
·
verified ·
1 Parent(s): 9a3ea39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -22,7 +22,7 @@ def respond(message):
22
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
23
  response = response.split("### Assistant:")[-1].strip()
24
  if "### Human:" in response:
25
- response = response.split("### Human:")[1].strip()
26
  return response
27
 
28
  banner = gr.HTML("""
@@ -42,9 +42,9 @@ with gr.Blocks(theme=gr.Theme.from_hub('gradio/monochrome')) as demo:
42
  clear = gr.Button("Clear chat")
43
  submit = gr.Button("Send message")
44
 
45
- def user_input(user_message):
46
  response = respond(user_message)
47
- return "", response
48
 
49
  msg.submit(user_input, [msg, chatbot], [msg, chatbot], queue=False)
50
  clear.click(lambda: None, None, chatbot, queue=False)
 
22
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
23
  response = response.split("### Assistant:")[-1].strip()
24
  if "### Human:" in response:
25
+ response = response.split("### Human:")[0].strip()
26
  return response
27
 
28
  banner = gr.HTML("""
 
42
  clear = gr.Button("Clear chat")
43
  submit = gr.Button("Send message")
44
 
45
+ def user_input(user_message, history):
46
  response = respond(user_message)
47
+ return "", history + [[user_message, response]]
48
 
49
  msg.submit(user_input, [msg, chatbot], [msg, chatbot], queue=False)
50
  clear.click(lambda: None, None, chatbot, queue=False)