tweaks

app.py

@@ -31,8 +31,8 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column(scale=1):
                 input_image = gr.Image(type="pil")
-            with gr.Column(scale=3):
-                output_image = gr.Gallery()
+            with gr.Column(scale=2):
+                output_image = gr.Gallery(height=600)
 
 
         @input_image.upload(inputs=input_image, outputs=output_image)
@@ -41,7 +41,7 @@ with gr.Blocks() as demo:
             inputs = extractor(images=resized_image, return_tensors="pt")
             outputs = model(**inputs)
             embed = outputs.last_hidden_state[0][0]
-            result = index.query(vector=embed.tolist(), top_k=3)
+            result = index.query(vector=embed.tolist(), top_k=4)
             return [dataset["train"][int(vector.id[3:])]["image"] for vector in result]
 
     with gr.Tab("Advanced"):