Update app.py

app.py

@@ -280,9 +280,9 @@ with gr.Blocks(css=css) as demo:
                         )
                         init_image = gr.Image(label="Input Image", type="pil")
                         with gr.Row():
-                            img2img_run_button = gr.Button("Generate", variant="primary", scale=1, full_width=True)
+                            img2img_run_button = gr.Button("Generate", variant="primary")
                     with gr.Column(scale=1):
-                        img2img_output = gr.Gallery(label="Result", elem_id="gallery", show_label=False, height=512, width=512).style(grid=[2, 2], height="auto")
+                        img2img_output = gr.Gallery(label="Result", elem_id="gallery").style(grid=[2, 2], height="auto")
                 with gr.Accordion("Advanced options", open=False):
                     with gr.Row():
                         img2img_use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)