Update app.py

app.py

@@ -250,7 +250,33 @@ 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)