Update app.py

app.py

@@ -91,9 +91,8 @@ def predict(input, max_length, top_p, temperature, history):
 
         yield postprocess(response), history
 
-    response = myChatMessage(role=response_piece.role, content=response_piece.content)
-    history.append(response)
-    yield response.content, history
+    history.append(myChatMessage(role=response_piece.role, content=response))
+    yield response, history
 
 
 def reset_user_input():