Update app.py

app.py

@@ -49,10 +49,9 @@ chat_history = []
 
 def chat_output(inputs):
     output = chat_interface(inputs)
-    answer = output.split('AI:')
-    if(len(answer) == 2):
-        chat_history.append((inputs, answer[1]))
-    chat_history.append((inputs, answer[1]))
+    answer_after_input = output.split('AI:')
+    answer = answer_after_input.split('Human:')
+    chat_history.append((inputs, answer[0]))
     return chat_history
 
 template = """You are an AI having conversation with a human.