Update handler.py

handler.py

@@ -19,20 +19,20 @@ class EndpointHandler():
         Return:
             A :obj:`list`: une liste permettant de passer les embedding 
         """
+        
+        serializable_results = {}
         inputs_request = data.pop("inputs", data)
-
-        # decode base64 image to PIL
-        image = Image.open(BytesIO(base64.b64decode(inputs_request['image'])))
-        text = inputs_request['text']
-        if text is not None:
+        
+        if 'text' in inputs_request:
+            text = inputs_request['text']
             text_embedding = self.model.encode_text(text)
-        if image is not None:
+            serializable_results['text_embedding'] = text_embedding.tolist() if isinstance(text_embedding, np.ndarray) else text_embedding
+            
+        
+        if 'image' in inputs_request:
+            image = Image.open(BytesIO(base64.b64decode(inputs_request['image'])))
             image_embedding = self.model.encode_image(image)
-        # Convert embeddings to lists of floats
-        serializable_results = {
-            'text_embedding': (text_embedding.tolist() if isinstance(text_embedding, np.ndarray) else text_embedding) if text_embedding is not None else [],
-            'image_embedding': (image_embedding.tolist() if isinstance(image_embedding, np.ndarray) else image_embedding) if image_embeddingis not None else []
-        }
-
+            serializable_results['image_embedding'] = image_embedding.tolist() if isinstance(image_embedding, np.ndarray) else image_embedding
+            
         return serializable_results