Update app.py

app.py

@@ -356,7 +356,7 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                             value=default_model_name,
                             label="Model",
                         )
-
+                        paste_back = gr.Checkbox(True, label="Paste back original")
                         loading_message = gr.Label(label="Status", value="", visible=False)  # Added loading message label
 
                 with gr.Row():
@@ -367,9 +367,8 @@ 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")
+                
 
                 use_as_input_button = gr.Button("Use as Input Image", visible=False)