Update app.py

app.py

@@ -292,16 +292,20 @@ with gr.Blocks() as demo:
             with gr.Row():
                 with gr.Column() as c1:
                     image_1 = gr.Image(interactive=False)
+                    image_1.style(height=256, width=256)
                     image_1_button = gr.Button("Select 1").style(full_width=True)
                 with gr.Column() as c2:
                     image_2 = gr.Image(interactive=False)
+                    image_2.style(height=256, width=256)
                     image_2_button = gr.Button("Select 2").style(full_width=True)
             with gr.Row():
                 with gr.Column() as c3:
                     image_3 = gr.Image(interactive=False)
+                    image_3.style(height=256, width=256)
                     image_3_button = gr.Button("Select 3").style(full_width=True)
                 with gr.Column() as c4:
                     image_4 = gr.Image(interactive=False)
+                    image_4.style(height=256, width=256)
                     image_4_button = gr.Button("Select 4").style(full_width=True)         
 
     start_button.click(