Update handler.py

handler.py

@@ -34,4 +34,11 @@ class EndpointHandler:
             predictions = self.pipeline(inputs)
         
         # postprocess the prediction
-        return [{'next_chat_turn': e[0]["generated_text"][-1]} for e in predictions]
+        results = []
+        for e in predictions:
+            e_turn = e[0]["generated_text"][-1]
+            results.append({
+                'next_chat_turn': e_turn,
+                'next_chat_text': e_turn['content'],
+            })
+        return results