Update app.py

app.py

@@ -22,14 +22,15 @@ links = np.load('data.npy', allow_pickle=True)
 
 
 
-def get_html(url_list, height=150):
+def get_html(url_list, height=180):
     html = "<div style='margin-top: 20px; max-width: 1200px; display: flex; flex-wrap: wrap; justify-content: space-evenly'>"
     for url in url_list:
         html2 = f"<img style='height: {height}px; margin: 5px' src='{escape(url)}'>"  
         html = html + html2
     html += "</div>"
     return html
-               
+ 
+@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