update

app.py

@@ -30,16 +30,16 @@ for name in names:
         continue
     subprocess.run(shlex.split(command), cwd='ControlNet/annotator/ckpts/')
 
-from gradio_canny2image import create_demo as create_demo_canny
-from gradio_depth2image import create_demo as create_demo_depth
-from gradio_fake_scribble2image import create_demo as create_demo_fake_scribble
-from gradio_hed2image import create_demo as create_demo_hed
-from gradio_hough2image import create_demo as create_demo_hough
-from gradio_normal2image import create_demo as create_demo_normal
-from gradio_pose2image import create_demo as create_demo_pose
-from gradio_scribble2image import create_demo as create_demo_scribble
-from gradio_scribble2image_interactive import \
-    create_demo as create_demo_scribble_interactive
+#from gradio_canny2image import create_demo as create_demo_canny
+#from gradio_depth2image import create_demo as create_demo_depth
+#from gradio_fake_scribble2image import create_demo as create_demo_fake_scribble
+#from gradio_hed2image import create_demo as create_demo_hed
+#from gradio_hough2image import create_demo as create_demo_hough
+#from gradio_normal2image import create_demo as create_demo_normal
+#from gradio_pose2image import create_demo as create_demo_pose
+#from gradio_scribble2image import create_demo as create_demo_scribble
+#from gradio_scribble2image_interactive import \
+#    create_demo as create_demo_scribble_interactive
 from gradio_seg2image import create_demo as create_demo_seg
 from model import Model
 
@@ -62,27 +62,27 @@ model = Model()
 with gr.Blocks(css='style.css') as demo:
     gr.Markdown(DESCRIPTION)
     with gr.Tabs():
-        with gr.TabItem('Canny'):
-            create_demo_canny(model.process_canny, max_images=MAX_IMAGES)
-        with gr.TabItem('Hough'):
-            create_demo_hough(model.process_hough, max_images=MAX_IMAGES)
-        with gr.TabItem('HED'):
-            create_demo_hed(model.process_hed, max_images=MAX_IMAGES)
-        with gr.TabItem('Scribble'):
-            create_demo_scribble(model.process_scribble, max_images=MAX_IMAGES)
-        with gr.TabItem('Scribble Interactive'):
-            create_demo_scribble_interactive(
-                model.process_scribble_interactive, max_images=MAX_IMAGES)
-        with gr.TabItem('Fake Scribble'):
-            create_demo_fake_scribble(model.process_fake_scribble,
-                                      max_images=MAX_IMAGES)
-        with gr.TabItem('Pose'):
-            create_demo_pose(model.process_pose, max_images=MAX_IMAGES)
+       #with gr.TabItem('Canny'):
+       #    create_demo_canny(model.process_canny, max_images=MAX_IMAGES)
+       #with gr.TabItem('Hough'):
+       #    create_demo_hough(model.process_hough, max_images=MAX_IMAGES)
+       #with gr.TabItem('HED'):
+       #    create_demo_hed(model.process_hed, max_images=MAX_IMAGES)
+       #with gr.TabItem('Scribble'):
+       #    create_demo_scribble(model.process_scribble, max_images=MAX_IMAGES)
+       #with gr.TabItem('Scribble Interactive'):
+       #    create_demo_scribble_interactive(
+       #        model.process_scribble_interactive, max_images=MAX_IMAGES)
+       #with gr.TabItem('Fake Scribble'):
+       #    create_demo_fake_scribble(model.process_fake_scribble,
+       #                              max_images=MAX_IMAGES)
+       #with gr.TabItem('Pose'):
+       #    create_demo_pose(model.process_pose, max_images=MAX_IMAGES)
         with gr.TabItem('Segmentation'):
             create_demo_seg(model.process_seg, max_images=MAX_IMAGES)
-        with gr.TabItem('Depth'):
-            create_demo_depth(model.process_depth, max_images=MAX_IMAGES)
-        with gr.TabItem('Normal map'):
-            create_demo_normal(model.process_normal, max_images=MAX_IMAGES)
+        #with gr.TabItem('Depth'):
+        #    create_demo_depth(model.process_depth, max_images=MAX_IMAGES)
+        #with gr.TabItem('Normal map'):
+        #    create_demo_normal(model.process_normal, max_images=MAX_IMAGES)
 
 demo.queue(api_open=False).launch()