Update app.py

app.py

@@ -239,10 +239,8 @@ with gr.Blocks(css=css) as demo:
                         cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
                         seed = gr.Slider(label="Seed", minimum=-1, maximum=10000000, value=-1)
     
-                with gr.Column():
-                             text_button = gr.Button("Генерация", variant='primary', elem_id="generate")
-                with gr.Column(scale=2):
-                    image_output = gr.Image()
+                text_button = gr.Button("Генерация", variant='primary', elem_id="generate")
+                image_output = gr.Image()
     
             text_button.click(txt2img, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed], outputs=image_output)
         
@@ -278,10 +276,8 @@ with gr.Blocks(css=css) as demo:
                         i2i_seed = gr.Slider(label="Seed", minimum=-1, maximum=10000000, value=-1)
 
                 
-                with gr.Column():
-                    i2i_text_button = gr.Button("Генерация", variant='primary', elem_id="generate")
-                with gr.Column(scale=2):
-                    i2i_image_output = gr.Image()
+                i2i_text_button = gr.Button("Генерация", variant='primary', elem_id="generate")
+                i2i_image_output = gr.Image()
 
             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)
 demo.queue(concurrency_count=64, max_size=80, api_open=False).launch(max_threads=256)