Update app.py

app.py

@@ -342,30 +342,33 @@ with gr.Blocks(css=css, fill_height=True) as demo:
     with gr.Tabs():
         with gr.TabItem("Inpaint"):
             with gr.Column():
-                with gr.Row(equal_height=True):
+                with gr.Row():
                     with gr.Column():
                         prompt = gr.Textbox(
                             label="Prompt",
                             info="Describe what to inpaint the mask with",
                             lines=3,
                         )
-                        input_image = gr.ImageMask(
-                            type="pil", label="Input Image", layers=True
-                        )
+
                     with gr.Column():
                         model_selection = gr.Dropdown(
                             choices=list(MODELS.keys()),
                             value=default_model_name,
                             label="Model",
                         )
-                        loading_message = gr.Label(label="Status", value="", visible=False)  # Added loading message label
-                        result = ImageSlider(
-                            interactive=False,
-                            label="Generated Image",
-                        )
                         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):
+                    input_image = gr.ImageMask(
+                        type="pil", label="Input Image", layers=True
+                    )
+                    result = ImageSlider(
+                        interactive=False,
+                        label="Generated Image",
+                    )
 
 
                 use_as_input_button = gr.Button("Use as Input Image", visible=False)