Update handler.py

handler.py

@@ -1,4 +1,4 @@
-from transformers import CLIPProcessor, CLIPModel, pipeline
+from transformers import pipeline
 from PIL import Image
 import requests
 from io import BytesIO
@@ -7,12 +7,8 @@ from typing import Dict, List, Any
 
 class EndpointHandler():
     def __init__(self, path=""):
-        # Utilitza el nom correcte del model per carregar-lo des de Hugging Face Hub
-        self.model = CLIPModel.from_pretrained("hf-hub:rroset/CLIP-ViT-B-32-laion2B-s34B-b79K")
-        self.processor = CLIPProcessor.from_pretrained("hf-hub:rroset/CLIP-ViT-B-32-laion2B-s34B-b79K")
-
-        # Crea la pipeline de classificació d'imatges zero-shot
-        self.classifier = pipeline("zero-shot-image-classification", model=self.model, processor=self.processor)
+        # Crea la pipeline de classificació d'imatges zero-shot amb el model específic
+        self.classifier = pipeline("zero-shot-image-classification", model="rroset/CLIP-ViT-B-32-laion2B-s34B-b79K")
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         image_input = data.get("inputs", None)
@@ -21,7 +17,7 @@ class EndpointHandler():
         if image_input is None or candidate_labels is None:
             raise ValueError("Image input or candidate labels not provided")
 
-        # Determina si l'input és una URL o una cadena base64
+        # Carregar la imatge, podria ser via URL o base64
         if image_input.startswith("http"):
             response = requests.get(image_input)
             image = Image.open(BytesIO(response.content))