Update app.py

app.py

@@ -346,6 +346,9 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                             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()),
@@ -353,18 +356,14 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                             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")
-                with gr.Row(fill_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)