Update handler.py

handler.py

@@ -11,20 +11,21 @@ class EndpointHandler():
         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)
-        candidate_labels = data.get("candidate_labels", None)
+        # Obté l'imatge en base64 i els paràmetres de les dades
+        image_base64 = data.get("inputs", None)
+        parameters = data.get("parameters", None)
 
-        if image_input is None or candidate_labels is None:
-            raise ValueError("Image input or candidate labels not provided")
+        if image_base64 is None or parameters is None:
+            raise ValueError("Input data or parameters not provided")
 
-        # 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))
-        else:
-            # Suposa que l'input és base64 i decodifica-la
-            image_data = base64.b64decode(image_input)
-            image = Image.open(BytesIO(image_data))
+        # Obté les etiquetes candidates dels paràmetres
+        candidate_labels = parameters.get("candidate_labels", None)
+        if candidate_labels is None:
+            raise ValueError("Candidate labels not provided")
+
+        # Decodifica la imatge des de base64
+        image_data = base64.b64decode(image_base64)
+        image = Image.open(BytesIO(image_data))
 
         # Realitza la classificació zero-shot
         results = self.classifier(images=image, candidate_labels=candidate_labels)