Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -83,7 +83,7 @@ def get_average_embedding(main_image_url, additional_image_urls):
|
|
| 83 |
|
| 84 |
if embeddings:
|
| 85 |
avg_embedding = np.mean(embeddings, axis=0)
|
| 86 |
-
return avg_embedding
|
| 87 |
else:
|
| 88 |
return None
|
| 89 |
|
|
|
|
| 83 |
|
| 84 |
if embeddings:
|
| 85 |
avg_embedding = np.mean(embeddings, axis=0)
|
| 86 |
+
return avg_embedding if isinstance(avg_embedding, np.ndarray) else avg_embedding
|
| 87 |
else:
|
| 88 |
return None
|
| 89 |
|