Update app.py

app.py

@@ -14,8 +14,10 @@ def predict(input, history=None):
     model.setState(history)
     model.loadContext(newctx=f"{input}\n\nBot: ")
     r = model.forward(number=100,stopStrings=["User: "])
+    rr = r["output"].split("<|endoftext|>")
+    rr = [(rr[i], rr[i+1]) for i in range(0, len(rr)-1, 2)]  # convert to tuples of list
     
-    return r["output"], r["state"]
+    return rr, r["state"]
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()