arman-aminian commited on
Commit
f885d9e
·
verified ·
1 Parent(s): 51358ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -10,7 +10,6 @@ image_embeddings = torch.load('image_embeddings.pt')
10
  image_links = np.load('image_links (1).npy', allow_pickle=True)
11
 
12
 
13
- @st.experimental_memo
14
  def image_search(query, top_k=10):
15
  with torch.no_grad():
16
  text_embedding = text_encoder(**text_tokenizer(query, return_tensors='pt')).pooler_output
 
10
  image_links = np.load('image_links (1).npy', allow_pickle=True)
11
 
12
 
 
13
  def image_search(query, top_k=10):
14
  with torch.no_grad():
15
  text_embedding = text_encoder(**text_tokenizer(query, return_tensors='pt')).pooler_output