Upload folder using huggingface_hub

.python-version

@@ -0,0 +1 @@
+3.9

handler.py

@@ -5,6 +5,8 @@ from typing import Dict, Any, List
 import requests
 import numpy as np
 from fashion_clip.fashion_clip import FashionCLIP
+from io import BytesIO
+
 
 
 class EndpointHandler:
@@ -23,30 +25,35 @@ class EndpointHandler:
             List[Dict[str, Any]]: The embeddings for the text and/or images.
         """
         # Extract text and images from the input data
-        texts = data.get("text", [])
-        images = data.get("image", [])
+        texts = data['inputs'].get("text", [])
+        images = data['inputs'].get("image", [])
+
 
         # Convert image URLs to PIL Images if needed
         images = [self._load_image(img) for img in images]
 
-        image_embeddings = self.fclip.encode_images(images, batch_size=32)
-        text_embeddings = self.fclip.encode_text(texts, batch_size=32)
-
-        image_embeddings = image_embeddings/np.linalg.norm(image_embeddings, ord=2, axis=-1, keepdims=True)
-        text_embeddings = text_embeddings/np.linalg.norm(text_embeddings, ord=2, axis=-1, keepdims=True)
+        results = {}
 
-        results = {
-            "image_embeddings": image_embeddings.tolist(),
-            "text_embeddings": text_embeddings.tolist()
-        }
+        if images:
+            image_embeddings = self.fclip.encode_images(images, batch_size=32)
+            image_embeddings = image_embeddings/np.linalg.norm(image_embeddings, ord=2, axis=-1, keepdims=True)
+            results["image_embeddings"] = image_embeddings.tolist()
+  
+        if texts:
+            text_embeddings = self.fclip.encode_text(texts, batch_size=32)
+            text_embeddings = text_embeddings/np.linalg.norm(text_embeddings, ord=2, axis=-1, keepdims=True)
+            results["text_embeddings"] = text_embeddings.tolist()
 
         return results
 
     def _load_image(self, img):
-        """Helper function to load an image from a URL, PIL Image, or numpy array."""
+        """Helper function to load an image from a URL, PIL Image, numpy array, or bytes."""
         if isinstance(img, str):
             # If the image is a URL
             img = Image.open(requests.get(img, stream=True).raw)
+        elif isinstance(img, bytes):
+            # If the image is in bytes
+            img = Image.open(BytesIO(img))
         elif isinstance(img, Image.Image):
             # If the image is already a PIL Image
             pass