Spaces:
Sleeping
Sleeping
Commit
·
e05a8b4
1
Parent(s):
4b3d8d5
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,10 @@ def search(image=None, query=None):
|
|
| 30 |
image_embed = np.array(pipe.encode(image))
|
| 31 |
|
| 32 |
if query is not None and image is not None:
|
| 33 |
-
|
|
|
|
|
|
|
|
|
|
| 34 |
elif image is not None:
|
| 35 |
embed = image_embed
|
| 36 |
else:
|
|
|
|
| 30 |
image_embed = np.array(pipe.encode(image))
|
| 31 |
|
| 32 |
if query is not None and image is not None:
|
| 33 |
+
text_embed_normalized = text_embed / np.linalg.norm(text_embed)
|
| 34 |
+
image_embed_normalized = image_embed / np.linalg.norm(image_embed)
|
| 35 |
+
embed = text_embed_normalized + image_embed_normalized
|
| 36 |
+
embed = embed / np.linalg.norm(embed) # we renormalize
|
| 37 |
elif image is not None:
|
| 38 |
embed = image_embed
|
| 39 |
else:
|