Update app.py

app.py

@@ -13,7 +13,7 @@ links = np.load('data.npy', allow_pickle=True)
 
 
 def get_html(url_list):
-    html = "<div style='margin-top: 50px; max-width: 1000px; display: flex; flex-wrap: wrap; justify-content: space-evenly'>"
+    html = "<div style='margin-top: 50px; max-width: 1100px; display: flex; flex-wrap: wrap; justify-content: space-evenly'>"
     for url in url_list:
         html2 = f"<img style='height: 200px; margin: 2px' src='{escape(url)}'>"  
         html = html + html2
@@ -21,7 +21,7 @@ def get_html(url_list):
     return html
  
                            
-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)
@@ -32,8 +32,7 @@ description = '''
 # Persian (fa) image search 
 - Enter your query and hit enter
 
-## Built with [CLIP-fa](https://github.com/sajjjadayobi/CLIPfa) model
-- Made with ❤️ in my basement
+Built with [CLIP-fa](https://github.com/sajjjadayobi/CLIPfa) model and 25k images from Unsplash
 '''