Update app.py

app.py

@@ -25,16 +25,13 @@ def get_nearest_64(gender, ethnicity, model, no):
     return image, list(zip(neighbor_images, neighbor_captions))
 
 with gr.Blocks() as demo:
+    gender = gr.Radio(gender_labels, label="Gender label")
+    ethnicity = gr.Radio(ethnicity_labels, label="Ethnicity label")
+    model = gr.Radio(models, label="Model")
+    no = gr.Radio(nos, label="Image number")
+    button = gr.Button(value="Get nearest neighbors")
     with gr.Row():
-        with gr.Column():
-            gender = gr.Radio(gender_labels, label="Gender label")
-            ethnicity = gr.Radio(ethnicity_labels, label="Ethnicity label")
-            model = gr.Radio(models, label="Model")
-            no = gr.Radio(nos, label="Image number")
-            button = gr.Button(value="Get nearest neighbors")
-        with gr.Column():
-            image = gr.Image(shape=(64,64))
-    gallery = gr.Gallery().style(grid=[2,4])
-    
+        image = gr.Image()
+        gallery = gr.Gallery().style(grid=8)
     button.click(get_nearest_64, inputs=[gender, ethnicity, model, no], outputs=[image, gallery])
 demo.launch()