Update app.py

app.py

@@ -178,7 +178,6 @@ with demo:
     # Combined mask controls
     bt_masks = gr.Button("Compute Masks")
     with gr.Row():
-        mask_image = gr.Image(type='numpy', label="Diffusion Mask")
         masked_image = gr.Image(type='pil', label="Masked Image")
     mask_storage = gr.State()
 
@@ -200,8 +199,9 @@ with demo:
             steps_slider = gr.Slider(minimum=1, maximum=100, value=50, label="Inference Steps")
             guidance_slider = gr.Slider(minimum=0.0, maximum=50.0, value=7.5, step=0.1, label="Guidance Scale")
             bt_diffusion = gr.Button("Run Diffusion")
+    mask_image = gr.Image(type='numpy', label="Diffusion Mask")
 
-        inpainted_image = gr.Image(type='pil', label="Inpainted Image")
+    inpainted_image = gr.Image(type='pil', label="Inpainted Image")
 
     # TODO: saw a better way of handling many inputs online..
     # forgot where though