Update app.py

app.py

@@ -211,7 +211,32 @@ with gr.Blocks(css=css) as demo:
                     negative_prompt = gr.Textbox(placeholder="Negative Prompt", show_label=False, lines=3, value="")
                 with gr.Column():
                     text_button = gr.Button("Generate", variant='primary', elem_id="generate")
-                        
+                    
+            with gr.Row():
+                with gr.Column(scale=3):
+                    with gr.Tab("Generation"):
+                        with gr.Row():
+                            with gr.Column(scale=1):
+                                sampler = gr.Dropdown(value="DPM++ 2M Karras", show_label=True, label="Sampling Method", choices=prodia_client.list_samplers())
+                                
+                            with gr.Column(scale=1):
+                                steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=30, value=25, step=1)
+    
+                        with gr.Row():
+                            with gr.Column(scale=1):
+                                width = gr.Slider(label="Width", maximum=1024, value=512, step=8)
+                                height = gr.Slider(label="Height", maximum=1024, value=512, step=8)
+                            
+                            with gr.Column(scale=1):
+                                batch_size = gr.Slider(label="Batch Size", maximum=1, value=1)
+                                batch_count = gr.Slider(label="Batch Count", maximum=1, value=1)
+    
+                        cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
+                        seed = gr.Number(label="Seed", value=-1)
+
+                with gr.Column(scale=2):
+                    image_output = gr.Image(value="https://huggingface.co/spaces/salomonsky/fast/raw/main/image.jpg")
+    
             text_button.click(txt2img, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height,
                                                seed], outputs=image_output, concurrency_limit=64)
         
@@ -250,33 +275,7 @@ with gr.Blocks(css=css) as demo:
 
                 with gr.Column(scale=2):
                     i2i_image_output = gr.Image(value="https://huggingface.co/spaces/salomonsky/fast/raw/main/image.jpg")
-
-            
-            with gr.Row():
-                with gr.Column(scale=3):
-                    with gr.Tab("Generation"):
-                        with gr.Row():
-                            with gr.Column(scale=1):
-                                sampler = gr.Dropdown(value="DPM++ 2M Karras", show_label=True, label="Sampling Method", choices=prodia_client.list_samplers())
-                                
-                            with gr.Column(scale=1):
-                                steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=30, value=25, step=1)
     
-                        with gr.Row():
-                            with gr.Column(scale=1):
-                                width = gr.Slider(label="Width", maximum=1024, value=512, step=8)
-                                height = gr.Slider(label="Height", maximum=1024, value=512, step=8)
-                            
-                            with gr.Column(scale=1):
-                                batch_size = gr.Slider(label="Batch Size", maximum=1, value=1)
-                                batch_count = gr.Slider(label="Batch Count", maximum=1, value=1)
-    
-                        cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
-                        seed = gr.Number(label="Seed", value=-1)
-
-                with gr.Column(scale=2):
-                    image_output = gr.Image(value="https://huggingface.co/spaces/salomonsky/fast/raw/main/image.jpg")
-            
             i2i_text_button.click(img2img, inputs=[i2i_image_input, i2i_denoising, i2i_prompt, i2i_negative_prompt,
                                                    model, i2i_steps, i2i_sampler, i2i_cfg_scale, i2i_width, i2i_height,
                                                    i2i_seed], outputs=i2i_image_output, concurrency_limit=64)