Update handler.py

handler.py

@@ -81,8 +81,8 @@ class EndpointHandler():
         # process input
         inputs = data.pop("inputs", data)
         # process input text
-        prediction = predict(inputs['text'], self.model, self.tokenizer)
-        return {"generated_text": prediction}
+        prediction = predict(inputs, self.model, self.tokenizer)
+        return prediction