more search results

app.py

@@ -106,8 +106,8 @@ def drop_same_folder(indices):
 
 
 def features_to_gallery(features):
-    indices = annoy_index.get_nns_by_vector(features, n=100)
-    indices = drop_same_folder(indices)[:20]
+    indices = annoy_index.get_nns_by_vector(features, n=500)
+    indices = drop_same_folder(indices)[:50]
     top_urls = urls[indices]
     return top_urls.tolist(), indices
 
@@ -167,7 +167,7 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
         query_uploaded_image_button = gr.Button("Show similiar to uploaded")
 
     gallery = gr.Gallery(label="Images", show_label=False, elem_id="gallery"
-        ).style(columns=4, container=False)
+        ).style(columns=5, container=False)
 
     with gr.Row():
         filename_textbox = gr.Textbox("", show_label=False).style(container=False)