Update app.py

app.py

@@ -59,20 +59,13 @@ 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():
-            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])
+            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]
+
+            for m, o in zip(current_models, output):
+                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)