Update handler.py

handler.py

@@ -1,19 +1,12 @@
 from typing import Dict, List, Any
 from transformers import pipeline
-import holidays
-
 class EndpointHandler():
     def __init__(self, path=""):
+        # 初始化方法，加载模型
         self.pipeline = pipeline("text-classification", model=path)
-        self.holidays = holidays.US()
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        inputs = data.pop("inputs", "")
-        date = data.pop("date", None)
-
-        # 根据日期判断是否是假期
-        if date and date in self.holidays:
-            return [{"label": "happy", "score": 1}]
-
+        # 处理输入数据并返回预测结果
+        inputs = data.get("inputs", "")
         prediction = self.pipeline(inputs)
         return prediction