Update handler.py

handler.py

@@ -3,15 +3,11 @@ from transformers import pipeline
 
 class EndpointHandler():
     def __init__(self, path=""):
-        # 初始化对话模型
-        self.pipeline = pipeline("conversational", model=path)
+        # 初始化方法，加载模型
+        self.pipeline = pipeline("text-classification", model=path)
 
-    def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
-        # 从请求数据中获取输入文本
+    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+        # 处理输入数据并返回预测结果
         inputs = data.get("inputs", "")
-        
-        # 使用对话模型生成响应
-        conversation = self.pipeline(inputs)
-        
-        # 返回模型的响应
+        prediction = self.pipeline(inputs)
         return {"response": conversation.generated_responses[0]}