Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|