Update app.py

app.py

@@ -25,13 +25,15 @@ def get_html(url_list, height=224):
         html2 = f"<img style='height: {height}px; margin: 5px' src='{escape(url)}'>"  
         html = html + html2
     html += "</div>"
-    return html
+    return HTML
 
+def compute_embeddings(query):
+    return text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output
 
-#@st.cache(show_spinner=False)
+@st.cache(show_spinner=False)
 def image_search(query, top_k=8):
     with torch.no_grad():
-        text_embedding = text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output
+        text_embedding = compute_embeddings(query)
     values, indices = torch.cosine_similarity(text_embedding, image_embeddings).sort(descending=True)
     return [links[i] for i in indices[:top_k]]