Update handler.py

handler.py

@@ -37,4 +37,4 @@ class EndpointHandler():
          )
         output_strs = self.tokenizer.batch_decode(outputs, skip_special_tokens=True)
         
-        return output_strs
+        return {"generated_text": output_strs}