Update handler.py

handler.py

@@ -5,7 +5,7 @@ class EndpointHandler:
     def __init__(self, path=""):
         # Load the optimized model
         tokenizer = AutoTokenizer.from_pretrained(path)
-        # Create inference pipeline
+        # Create inference pipeline for text classification
         self.pipeline = pipeline("text-classification", model=path, tokenizer=tokenizer)
 
     def __call__(self, data: str) -> List[List[Dict[str, float]]]:
@@ -17,8 +17,8 @@ class EndpointHandler:
             A list containing the prediction results:
             A list of one list, e.g., [[{"label": "LABEL", "score": 0.99}]]
         """
-        # Directly pass the input data to the pipeline without any type check
-        prediction = self.pipeline(data)
+        # Pass the data as `text` directly
+        prediction = self.pipeline(text=data)
 
         # Return the prediction result
         return prediction