Update app.py

app.py

@@ -28,7 +28,9 @@ with gr.Blocks() as demo:
     ethnicity = gr.Radio(ethnicity_labels, label="Ethnicity label")
     model = gr.Radio(models, label="Model")
     no = gr.Radio(nos, label="Image number")
-
+    with gr.Row():
+        image = gr.Image(shape=(64,64))
+        gallery = gr.Gallery().style(grid=8, height="auto")
     button = gr.Button(value="Get nearest neighbors")
-    button.click(get_nearest_64, inputs=[gender, ethnicity, model, no], outputs=[gr.Image(), gr.Gallery().style(grid=[8], height="auto")])
+    button.click(get_nearest_64, inputs=[gender, ethnicity, model, no], outputs=[image, gallery])
 demo.launch()