Update app.py

app.py

@@ -367,9 +367,9 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                         interactive=False,
                         label="Generated Image",
                     )
-                    with gr.Row():
-                        run_button = gr.Button("Generate")
-                        paste_back = gr.Checkbox(True, label="Paste back original")
+            with gr.Row():
+                run_button = gr.Button("Generate")
+                paste_back = gr.Checkbox(True, label="Paste back original")
 
                 use_as_input_button = gr.Button("Use as Input Image", visible=False)