Update handler.py

handler.py

@@ -54,4 +54,4 @@ class EndpointHandler:
         assistant_answer_start = generated_text[0].find("ASSISTANT:") + len("ASSISTANT:")
         assistant_answer = generated_text[0][assistant_answer_start:].strip()
 
-        return {"generated_text": assistant_answer}
+        return assistant_answer