Update handler.py

handler.py

@@ -9,5 +9,8 @@ class EndpointHandler():
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         # 处理输入数据并返回预测结果
         inputs = data.get("inputs", "")
-        prediction = self.pipeline(inputs)
+          # 使用对话模型生成响应
+        conversation = self.pipeline(inputs)
+        
+        # 返回模型的响应
         return {"response": conversation.generated_responses[0]}