Upload app.py

app.py

@@ -5,8 +5,7 @@ from annotator.util import resize_image, HWC3
 
 DESCRIPTION = '# ControlNet v1.1 Annotators (that runs on cpu only)'
 DESCRIPTION += '\n<p>This app generates Control Image for Mochi Diffusion&apos;s ControlNet.</p>'
-DESCRIPTION += '\n<p>HEIC image is not converted. Please use PNG or JPG image.'
-DESCRIPTION += '\n&nbsp;HEICの画像は変換できません。 PNGまたはJPGの画像を使ってください。</p>'
+DESCRIPTION += '\n<p>HEIC image is not converted. Please use PNG or JPG image.</p>'
 
 
 model_canny = None
@@ -211,6 +210,15 @@ def color_shuffler(img, res):
     result = model_color_shuffler(img)
     return [result]
 
+model_inpaint = None
+
+
+def inpaint(img, res):
+    image = resize_image(img, res)
+    color = HWC3(image["image"])
+    alpha = image["mask"][:, :, 0:1]
+    result = np.concatenate([color, alpha], axis=2)
+    return [result]
 
 block = gr.Blocks().queue()
 with block:
@@ -228,6 +236,18 @@ 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")
+            resolution = gr.Slider(label="resolution", minimum=256, maximum=1024, value=512, step=64)
+            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])
+
     gr.Markdown("<hr>")
     with gr.Row():
         gr.Markdown("## HED Edge&nbsp;&quot;SoftEdge&quot;")