try having the gallery outside of the row

app.py

@@ -429,9 +429,12 @@ A tool for exploring CLIP embedding space.
             n_samples = gr.Slider(1, 16, value=4, step=1, label="Number images")
         with gr.Column(scale=3, min_width=200):
             submit = gr.Button("Search embedding space")
-    with gr.Row():
-        output = gr.Gallery(label="Closest images in Laion 5b using kNN", show_label=True)\
-            .style(grid=[4,4], height="auto")
+    # with gr.Row():
+    #     output = gr.Gallery(label="Closest images in Laion 5b using kNN", show_label=True)\
+    #         .style(grid=[4,4], height="auto")
+    output = gr.Gallery(label="Closest images in Laion 5b using kNN", show_label=True)\
+        .style(grid=[4,4], height="auto")
+
 
     embedding_base64s_state = gr.State(value=[None for i in range(max_tabs)])
     embedding_power_state = gr.State(value=[1. for i in range(max_tabs)])