Update handler.py

handler.py

@@ -2,18 +2,18 @@ from typing import Dict, List, Any
 from transformers import pipeline
 
 class EndpointHandler():
-    def __init__(self, path="zemuwen/14B_lora_ze"):
-        # 初始化方法，加载文本分类模型
+     def __init__(self, path="zemuwen/14B_lora_ze"):
+        # 初始化方法，加载问答模型
         self.pipeline = pipeline("question-answering", model=path)
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        # 处理输入数据并返回预测结果
-        inputs = data.get("inputs", "")
+        # 确保输入数据包含 "question" 和 "context"
+        question = data.get("question", "")
+        context = data.get("context", "")
         
-        # 使用分类模型生成响应
-        results = self.pipeline(inputs)
+        # 使用问答模型生成响应
+        results = self.pipeline(question=question, context=context)
         
         # 返回模型的响应
         return results
 
-