Update app.py

app.py

@@ -36,9 +36,8 @@ def get_html(url_list, height=224):
 
 st.cache(show_spinner=False)
 def image_search(query, top_k=8):
-    torch.no_grad():
-        text_embedding = text_encoder(
-            **tokenizer(query, return_tensors='pt')).pooler_output
+    with torch.no_grad():
+        text_embedding = text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output
     values, indices = torch.cosine_similarity(text_embedding, image_embeddings).sort(descending=True)
     return [(df.iloc[i]['path'], df.iloc[i]['link']) for i in indices[:top_k]]