update handler and requirements.txt

handler.py

@@ -13,10 +13,16 @@ class EndpointHandler(object):
             device_map="auto",
         )
 
-    def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
+    def __call__(self, data: Dict[str, Any]) -> List[List[Dict[str, float]]]:
+        """
+        :param data:
+            inputs: message format
+            parameters: parameters for the pipeline
+        :return:
+        """
         inputs = data.pop("inputs")
 
-        if parameters:= data.pop("parameters", None):
+        if parameters:= data.pop("parameters") is not None:
             outputs = self.pipeline(
                 inputs,
                 **parameters

requirements.txt

@@ -1,2 +1 @@
-transformers[sklearn,sentencepiece,audio,vision,sentencepiece]==4.46.1
-huggingface-inference-toolkit==0.5.2
+transformers[sklearn,sentencepiece,audio,vision,sentencepiece]==4.46.1