Update handler.py

handler.py

@@ -34,4 +34,4 @@ class EndpointHandler():
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         res = self.generate_text("Explain to me the difference between nuclear fission and fusion.")
-        return res[0]["generated_text"]
+        return res