SajjadAyoubi commited on
Commit
36ee6e5
·
1 Parent(s): e6c4b07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def get_html(url_list):
25
  html += "</div>"
26
  return html
27
 
28
-
29
  def image_search(query, top_k=10):
30
  with torch.no_grad():
31
  text_embedding = text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output
 
25
  html += "</div>"
26
  return html
27
 
28
+ @st.experimental_memo
29
  def image_search(query, top_k=10):
30
  with torch.no_grad():
31
  text_embedding = text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output