LDanielBlueway commited on
Commit
c167b01
·
verified ·
1 Parent(s): b607f57

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -30,8 +30,8 @@ class EndpointHandler():
30
  image_embedding = self.model.encode_image(image)
31
  # Convert embeddings to lists of floats
32
  serializable_results = {
33
- 'text_embedding': text_embedding.tolist() if isinstance(text_embedding, np.ndarray) else text_embedding,
34
- 'image_embedding': image_embedding.tolist() if isinstance(image_embedding, np.ndarray) else image_embedding
35
  }
36
 
37
  return serializable_results
 
30
  image_embedding = self.model.encode_image(image)
31
  # Convert embeddings to lists of floats
32
  serializable_results = {
33
+ 'text_embedding': (text_embedding.tolist() if isinstance(text_embedding, np.ndarray) else text_embedding) if text_embedding is not None else [],
34
+ 'image_embedding': (image_embedding.tolist() if isinstance(image_embedding, np.ndarray) else image_embedding) if image_embeddingis not None else []
35
  }
36
 
37
  return serializable_results