Upload app.py

app.py

@@ -141,14 +141,17 @@ def lineart_anime(img, res):
 model_lineart = None
 
 
-def lineart(img, res, coarse=False):
+def lineart(img, res, coarse=False, invert=True):
     img = resize_image(HWC3(img), res)
     global model_lineart
     if model_lineart is None:
         from annotator.lineart import LineartDetector
         model_lineart = LineartDetector()
 #    result = model_lineart(img, coarse)
-    result = cv2.bitwise_not(model_lineart(img, coarse))
+    if invert is True
+        result = cv2.bitwise_not(model_lineart(img, coarse))
+    else
+        result = model_lineart(img, coarse)    
     return [result]
 
 
@@ -318,11 +321,12 @@ with block:
         with gr.Column():
             input_image = gr.Image(source='upload', type="numpy")
             coarse = gr.Checkbox(label='Using coarse model', value=False)
+            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, inputs=[input_image, resolution, coarse], outputs=[gallery])
+    run_button.click(fn=lineart, inputs=[input_image, resolution, coarse, invert], outputs=[gallery])
 
 #    with gr.Row():
 #        gr.Markdown("## Uniformer Segmentation")