Update app.py

app.py

@@ -59,18 +59,20 @@ with gr.Blocks() as demo:
         """
         )
 
+        # Layout for 3 columns and 2 rows
         output = [gr.Image(label=m, min_width=480) for m in default_models]
         current_models = [gr.Textbox(m, visible=False) for m in default_models]
 
         with gr.Row():
-            for i in range(0, 6, 3):
-                with gr.Column():
-                    for j in range(3):
-                        if i + j < len(output):
-                            o = output[i + j]
-                            m = current_models[i + j]
-                            gen_event = gen_button.click(gen_fn, [m, txt_input, neg_prompt, image_style], o, concurrency_limit=10)
-                            stop_button.click(lambda s: gr.update(interactive=False), None, stop_button, cancels=[gen_event])
+            with gr.Column():
+                for i in range(0, 6, 3):
+                    with gr.Row():
+                        for j in range(3):
+                            if i + j < len(output):
+                                o = output[i + j]
+                                m = current_models[i + j]
+                                gen_event = gen_button.click(gen_fn, [m, txt_input, neg_prompt, image_style], o, concurrency_limit=10)
+                                stop_button.click(lambda s: gr.update(interactive=False), None, stop_button, cancels=[gen_event])
 
         with gr.Accordion('Model selection'):
             model_choice = gr.CheckboxGroup(models, label=f'Choose up to {num_models} different models from the best available!', value=default_models, interactive=True)