Update to latest text encoder
Browse files
app.py
CHANGED
@@ -103,13 +103,13 @@ def search(model_name: str, img_input, str_input: str, n_neighbours: int):
|
|
103 |
if str_input == "":
|
104 |
embeddings = siglip.get_siglip_image_embedding(
|
105 |
img_input,
|
106 |
-
model_name="smilingwolf/
|
107 |
fmt="embeddings",
|
108 |
)
|
109 |
else:
|
110 |
embeddings = siglip.get_siglip_text_embedding(
|
111 |
str_input,
|
112 |
-
model_name="smilingwolf/
|
113 |
fmt="embeddings",
|
114 |
)
|
115 |
|
|
|
103 |
if str_input == "":
|
104 |
embeddings = siglip.get_siglip_image_embedding(
|
105 |
img_input,
|
106 |
+
model_name="smilingwolf/siglip_swinv2_base_2025_02_13_22h42m03s",
|
107 |
fmt="embeddings",
|
108 |
)
|
109 |
else:
|
110 |
embeddings = siglip.get_siglip_text_embedding(
|
111 |
str_input,
|
112 |
+
model_name="smilingwolf/siglip_swinv2_base_2025_02_13_22h42m03s",
|
113 |
fmt="embeddings",
|
114 |
)
|
115 |
|