Update handler.py
Browse files- 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
|