Update src/app.py

src/app.py

@@ -92,7 +92,7 @@ def get_neighbors(query_image, selected_descriptor, top_k=5):
         return []
 
 
-def load_cbir_dataset(datasetname, size):
+def load_cbir_dataset(datasetname, size=1000):
     pass
 
 # Define the Gradio Interface
@@ -125,7 +125,7 @@ with gr.Blocks() as demo:
         value=["huggan/CelebA-faces"]
     )
         btn_index = gr.Button(value="Switch Dataset")
-    btn_index.click(load_cbir_dataset, inputs=[dataset_dropdown], outputs=[dataset_with_embeddings])
+    btn_index.click(load_cbir_dataset, inputs=[dataset_dropdown])
     btn.click(get_neighbors, inputs=[image_input, checkboxes_descr], outputs=[gallery_output])
 
     # gr.Markdown(