Fix handler

handler.py

@@ -18,9 +18,9 @@ class EndpointHandler():
                 - "label": A string representing what the label/class is. There can be multiple labels.
                 - "score": A score between 0 and 1 describing how confident the model is for this label/class.
         """
-        inputs = data.pop("inputs", data)
-        inputs = self.tokenizer(inputs, padding="max_length")
-        preds = self.model(**inputs)
+        text = data.pop("inputs", data)
+        inputs = self.tokenizer(text, padding="max_length")
+        preds = self.model(input_ids=[inputs["input_ids"]])
 
         with open("label_binarizer.pkl", "rb") as f:
             label_binarizer = pickle.loads(f.read())