SajjadAyoubi commited on
Commit
99713ae
·
1 Parent(s): cda97bc

Update app.py

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