Spaces:
Runtime error
Runtime error
Commit
·
6cbb017
1
Parent(s):
260e0f1
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def compute_embeddings(query):
|
|
31 |
with torch.no_grad():
|
32 |
return text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output
|
33 |
|
34 |
-
|
35 |
def image_search(query, top_k=8):
|
36 |
text_embedding = compute_embeddings(query)
|
37 |
values, indices = torch.cosine_similarity(text_embedding, image_embeddings).sort(descending=True)
|
|
|
31 |
with torch.no_grad():
|
32 |
return text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output
|
33 |
|
34 |
+
st.cache(show_spinner=False)
|
35 |
def image_search(query, top_k=8):
|
36 |
text_embedding = compute_embeddings(query)
|
37 |
values, indices = torch.cosine_similarity(text_embedding, image_embeddings).sort(descending=True)
|