Upload app.py

app.py

@@ -4,7 +4,7 @@ import cv2
 from annotator.util import resize_image, HWC3
 
 DESCRIPTION = '# ControlNet v1.1 Annotators (that runs on cpu only)'
-DESCRIPTION += '\n<p>HEIC image are not converted. Please use PNG or JPG images.</p>'
+DESCRIPTION += '\n<p>HEIC image are not converted. Please use PNG or JPG image.</p>'
 
 
 model_canny = None
@@ -127,14 +127,18 @@ model_uniformer = None
 model_lineart_anime = None
 
 
-def lineart_anime(img, res):
+def lineart_anime(img, res, invert=True):
     img = resize_image(HWC3(img), res)
     global model_lineart_anime
     if model_lineart_anime is None:
         from annotator.lineart_anime import LineartAnimeDetector
         model_lineart_anime = LineartAnimeDetector()
 #    result = model_lineart_anime(img)
-    result = cv2.bitwise_not(model_lineart_anime(img))
+#    result = model_lineart(img, coarse)
+    if (invert):
+        result = cv2.bitwise_not(model_lineart(img, coarse))
+    else:
+        result = model_lineart(img, coarse)    
     return [result]
 
 
@@ -305,18 +309,19 @@ with block:
     run_button.click(fn=openpose, inputs=[input_image, resolution, hand_and_face], outputs=[gallery])
 
     with gr.Row():
-        gr.Markdown("## Lineart Anime (The converted image is color-inverted.)")
+        gr.Markdown("## Lineart Anime \n<p>Check Invert to use with Mochi Diffusion.")
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(source='upload', type="numpy")
+            invert = gr.Checkbox(label='Invert', value=True)
             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=lineart_anime, inputs=[input_image, resolution], outputs=[gallery])
+    run_button.click(fn=lineart_anime, inputs=[input_image, resolution, invert], outputs=[gallery])
 
     with gr.Row():
-        gr.Markdown("## Lineart (The converted image is color-inverted.)")
+        gr.Markdown("## Lineart \n<p>Check Invert to use with Mochi Diffusion.")
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(source='upload', type="numpy")