SajjadAyoubi commited on
Commit
6cbb017
·
1 Parent(s): 260e0f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)