modified postprocess

handler.py

@@ -34,4 +34,6 @@ class EndpointHandler:
             prediction = self.pipeline(messages)
             
         # postprocess the prediction
-        return prediction[0]["generated_text"]
+        assistant_output = prediction[0]["generated_text"][-1]
+        
+        return assistant_output["content"]