Spaces:
Runtime error
Runtime error
Commit
·
99713ae
1
Parent(s):
cda97bc
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def compute_embeddings(query):
|
|
35 |
with torch.no_grad():
|
36 |
return text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output
|
37 |
|
38 |
-
st.cache(show_spinner=False)
|
39 |
def image_search(query, top_k=8):
|
40 |
text_embedding = compute_embeddings(query)
|
41 |
values, indices = torch.cosine_similarity(text_embedding, image_embeddings).sort(descending=True)
|
|
|
35 |
with torch.no_grad():
|
36 |
return text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output
|
37 |
|
38 |
+
@st.cache(show_spinner=False)
|
39 |
def image_search(query, top_k=8):
|
40 |
text_embedding = compute_embeddings(query)
|
41 |
values, indices = torch.cosine_similarity(text_embedding, image_embeddings).sort(descending=True)
|