Update app.py

app.py

@@ -260,7 +260,7 @@ with block:
         gr.Markdown("## Inpaint \n<p>Mochi Diffusion v4.1で使えるようになりました")
     with gr.Row():
         with gr.Column():
-            input_image = gr.Sketchpad(type="numpy", height=512)
+            input_image = gr.ImageMask(sources="upload", type="numpy", height=640)
 #            input_image = gr.Image(source='upload', type="numpy", tool="sketch", height=512)
 #            resolution = gr.Slider(label="resolution", minimum=256, maximum=1024, value=512, step=64)
             invert = gr.Checkbox(label='Invert Mask', value=False)
@@ -283,7 +283,8 @@ with block:
 #            run_button = gr.Button(label="Run")
         with gr.Column():
             gallery = gr.Gallery(label="Generated images", show_label=False, height="auto")
-#            gallery = gr.Gallery(label="Generated images", show_label=False).style(height="auto")    run_button.click(fn=hed, inputs=[input_image, resolution], outputs=[gallery])
+#            gallery = gr.Gallery(label="Generated images", show_label=False).style(height="auto")
+    run_button.click(fn=hed, inputs=[input_image, resolution], outputs=[gallery])
 
     gr.Markdown("<hr>")
     with gr.Row():
@@ -463,6 +464,7 @@ with block:
             resolution = gr.Slider(label="resolution", minimum=256, maximum=1024, value=640, step=64)
             run_button = gr.Button("Run")
 #            run_button = gr.Button(label="Run")
+        with gr.Column():
             gallery = gr.Gallery(label="Generated images", show_label=False, height="auto")
 #            gallery = gr.Gallery(label="Generated images", show_label=False).style(height="auto")
     run_button.click(fn=oneformer_ade20k, inputs=[input_image, resolution], outputs=[gallery])