Update app.py

app.py

@@ -19,23 +19,28 @@ model = model.eval()
 #     history[-1][1] = response
 #     return history
 
-def predict(input, history=None):
-    if history is None:
-        history = []
-    response, history = model.chat(tokenizer, input, history)
-    return history, history
+# def predict(input, history=None):
+#     if history is None:
+#         history = []
+#     response, history = model.chat(tokenizer, input, history)
+#     return response, history
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
     clear = gr.Button("Clear")
 
+    def predict(input, history=None):
+        if history is None:
+            history = []
+        response, history = model.chat(tokenizer, input, history)
+        return response, history
+
     def user(user_message, history):
         return "", history + [[user_message, None]]
 
     def bot(msg, history):
-        # bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
-        bot_message = predict(msg, history)
+        bot_message, _ = predict(msg, history)
         history[-1][1] = ""
         for character in bot_message:
             history[-1][1] += character