Update app.py

app.py

@@ -48,8 +48,8 @@ def generate_images(
 # Create the Gradio blocks
 with gr.Blocks() as demo:
     with gr.Row(equal_height=False):
-        with gr.Group():
-            with gr.Column():
+        with gr.Column():
+            with gr.Group():
                 model_dropdown = gr.Dropdown(choices=list(models.keys()), value=model_list[0] if model_list else None, label="Model")
                 prompt = gr.Textbox(label="Prompt")
                 with gr.Accordion("Advanced", open=False):
@@ -60,9 +60,9 @@ with gr.Blocks() as demo:
                     width = gr.Slider(minimum=1024, maximum=2048, step=256, value=1024, label="Width")
                     num_images = gr.Slider(minimum=1, maximum=4, step=1, value=4, label="Number of Images")
                 generate_btn = gr.Button("Generate Image")
-            with gr.Column():
-                output_gallery = gr.Gallery(label="Generated Images", height=480, scale=1)
-            generate_btn.click(generate_images, inputs=[model_dropdown, prompt, negative_prompt, num_inference_steps, guidance_scale, height, width, num_images], outputs=output_gallery)
+        with gr.Column():
+            output_gallery = gr.Gallery(label="Generated Images", height=480, scale=1)
+        generate_btn.click(generate_images, inputs=[model_dropdown, prompt, negative_prompt, num_inference_steps, guidance_scale, height, width, num_images], outputs=output_gallery)
 
 demo.launch()