Update app.py

app.py

@@ -356,9 +356,7 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                             value=default_model_name,
                             label="Model",
                         )
-                        with gr.Row():
-                            run_button = gr.Button("Generate")
-                            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(equal_height=True):
@@ -369,7 +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")
 
                 use_as_input_button = gr.Button("Use as Input Image", visible=False)