Update app.py

app.py

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