Update app.py

app.py

@@ -289,24 +289,18 @@ with gr.Blocks() as demo:
         with gr.Row():
             prompt = gr.Markdown("")
         with gr.Blocks():
-            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)         
+            with gr.Column() as c1:
+                image_1 = gr.Image(interactive=False)
+                image_1_button = gr.Button("Select 1").style(full_width=True)
+            with gr.Column() as c2:
+                image_2 = gr.Image(interactive=False)
+                image_2_button = gr.Button("Select 2").style(full_width=True)
+            with gr.Column() as c3:
+                image_3 = gr.Image(interactive=False)
+                image_3_button = gr.Button("Select 3").style(full_width=True)
+            with gr.Column() as c4:
+                image_4 = gr.Image(interactive=False)
+                image_4_button = gr.Button("Select 4").style(full_width=True)         
 
     start_button.click(
         fn=start,