Update app.py

app.py

@@ -227,12 +227,13 @@ model_inpaint = None
 
 def inpaint(image, invert):
 #    image = resize_image(img, res)
-    color = HWC3(image["image"])
-    if(invert):
-        alpha = image["mask"][:, :, 0:1]
-    else:
-        alpha = 255 - image["mask"][:, :, 0:1]
-    result = np.concatenate([color, alpha], axis=2)
+#    color = HWC3(image["image"])
+#    if(invert):
+#        alpha = image["mask"][:, :, 0:1]
+#    else:
+#        alpha = 255 - image["mask"][:, :, 0:1]
+#    result = np.concatenate([color, alpha], axis=2)
+    result = image
     return [result]
 
 block = gr.Blocks().queue()
@@ -260,8 +261,7 @@ with block:
         gr.Markdown("## Inpaint \n<p>Mochi Diffusion v4.1で使えるようになりました")
     with gr.Row():
         with gr.Column():
-            input_image = gr.Image(label="Input Image", type="numpy", tools="sketch", height="auto")
-#            input_image = gr.ImageMask(sources="upload", type="numpy", height="auto")
+            input_image = gr.ImageMask(sources="upload", type="numpy", height="auto")
 #            input_image = gr.ImageEditor(sources="upload", type="numpy", height="auto", layers="False", brush=gr.Brush(colors=["#000000"]))
 #            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)