Upload app.py

app.py

@@ -217,7 +217,7 @@ model_inpaint = None
 def inpaint(image):
 #    image = resize_image(img, res)
     color = HWC3(image["image"])
-    alpha = image["mask"][:, :, 0:1]
+    alpha = HWC3(image["mask"][:, :, 0:1])
     result = np.concatenate([color, alpha], axis=2)
     return [result]
 
@@ -242,7 +242,7 @@ with block:
         gr.Markdown("## Inpaint \n<p>Mochi Diffusionの次バージョンで使えるようになるかもしれないので試作")
     with gr.Row():
         with gr.Column():
-            input_image = gr.Image(source='upload', type="numpy", tool="sketch")
+            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)
             run_button = gr.Button(label="Run")
         with gr.Column():