Update app.py

app.py

@@ -30,8 +30,8 @@ def get_html(url_list, height=180):
     html += "</div>"
     return html
  
-@st.cache(show_spinner=False)              
-def image_search(query, top_k=8):
+                           
+def image_search(query, top_k=10):
     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)