Update handler.py

handler.py

@@ -7,15 +7,13 @@ import base64
 
 class EndpointHandler():
     def __init__(self, path=""):
-        self.pipeline = pipeline("zero-shot-image-classification", device=0)
+        self.pipeline = pipeline("zero-shot-image-classification", model=path, device=0)
     
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         inputs = data.pop("inputs", {})
         parameters = data.pop("parameters", {})
         candidate_labels = parameters.pop("candidate_labels", [])
-        print("input: ", inputs)
-        print("input: ", parameters)
-        print("input: ", candidate_labels)
+        print("data: ", data)
 
         # decode base64 image to PIL
         image = Image.open(BytesIO(base64.b64decode(inputs)))