Upload app.py

app.py

@@ -214,10 +214,13 @@ def color_shuffler(img, res):
 model_inpaint = None
 
 
-def inpaint(image):
+def inpaint(image, invert):
 #    image = resize_image(img, res)
     color = HWC3(image["image"])
-    alpha = 255 - image["mask"][:, :, 0:1]
+    if(invert):
+        alpha = image["mask"][:, :, 0:1]
+    else:
+        alpha = 255 - image["mask"][:, :, 0:1]
     result = np.concatenate([color, alpha], axis=2)
     return [result]
 
@@ -239,16 +242,17 @@ with block:
 
     gr.Markdown("<hr>")
     with gr.Row():
-        gr.Markdown("## Inpaint \n<p>Mochi Diffusionの次バージョンで使えるようになるかもしれないので試作")
+        gr.Markdown("## Inpaint \n<p>Mochi Diffusionの次バージョンで使えるようになるかもしれないので試作中")
     with gr.Row():
         with gr.Column():
             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='Mask Invert', value=False)
             run_button = gr.Button(label="Run")
         with gr.Column():
             gallery = gr.Gallery(label="Generated images", show_label=False).style(height="auto")
 #    run_button.click(fn=inpaint, inputs=[input_image, resolution], outputs=[gallery])
-    run_button.click(fn=inpaint, inputs=[input_image], outputs=[gallery])
+    run_button.click(fn=inpaint, inputs=[input_image, invert], outputs=[gallery])
 
     gr.Markdown("<hr>")
     with gr.Row():