Update app.py

app.py

@@ -22,7 +22,7 @@ links = np.load('data.npy', allow_pickle=True)
 
 
 
-def get_html(url_list, height=300):
+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)}'>"  
@@ -31,7 +31,7 @@ def get_html(url_list, height=300):
     return html
  
                            
-def image_search(query, top_k=10):
+def image_search(query, top_k=8):
     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)