Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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:")[
|
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)
|