diff --git "a/app.py" "b/app.py" --- "a/app.py" +++ "b/app.py" @@ -30,16 +30,6 @@ 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_seg2image import create_demo as create_demo_seg from model import Model @@ -62,27 +52,81 @@ 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('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) + + + +desc = """ + + +
+ + + +