Update handler.py

handler.py

@@ -18,7 +18,8 @@ class EndpointHandler:
             A list of one list, e.g., [[{"label": "LABEL", "score": 0.99}]]
         """
         # Pass the data as `text` directly
-        prediction = self.pipeline(text=data)
+        inputs = data.pop("inputs", data)
+        prediction = self.pipeline(inputs)
 
         # Return the prediction result
         return prediction