Update app.py

app.py

@@ -289,18 +289,19 @@ with gr.Blocks() as demo:
         with gr.Row():
             prompt = gr.Markdown("")
         with gr.Blocks():
-            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)         
+            with gr.Row():
+                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,