Update handler.py

handler.py

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