Upload app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import cv2
-import numpy as np
 
 from annotator.util import resize_image, HWC3
 
@@ -211,18 +210,6 @@ def color_shuffler(img, res):
     result = model_color_shuffler(img)
     return [result]
 
-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)
-    return [result]
 
 block = gr.Blocks().queue()
 with block:
@@ -240,20 +227,6 @@ with block:
             gallery = gr.Gallery(label="Generated images", show_label=False).style(height="auto")
     run_button.click(fn=canny, inputs=[input_image, resolution, low_threshold, high_threshold], outputs=[gallery])
 
-    gr.Markdown("<hr>")
-    with gr.Row():
-        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='Invert Mask', 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, invert], outputs=[gallery])
-
     gr.Markdown("<hr>")
     with gr.Row():
         gr.Markdown("## HED Edge&nbsp;&quot;SoftEdge&quot;")