Update handler.py

handler.py

@@ -6,13 +6,13 @@ import base64
 from typing import Dict, List, Any
 
 class EndpointHandler():
-    def __init__(self, path="."):
-        # Carrega el model i el processor específics de CLIP
+    def __init__(self, path=""):
+        # Carrega el model i el processador específics de CLIP
         self.model = CLIPModel.from_pretrained(path)
         self.processor = CLIPProcessor.from_pretrained(path)
 
         # Crea la pipeline de classificació d'imatges zero-shot
-        self.classifier = pipeline("zero-shot-image-classification", model=self.model, tokenizer=self.processor)
+        self.classifier = pipeline("zero-shot-image-classification", model=self.model, processor=self.processor)
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         image_input = data.get("inputs", None)
@@ -26,7 +26,7 @@ class EndpointHandler():
             response = requests.get(image_input)
             image = Image.open(BytesIO(response.content))
         else:
-            # Suposa que l'input és base64 i decodifica-la
+            # Suposem que l'input és base64 i decodifica-la
             image_data = base64.b64decode(image_input)
             image = Image.open(BytesIO(image_data))