ysharma HF staff commited on
Commit
c9e5dc2
·
1 Parent(s): 00c70e2
Files changed (1) hide show
  1. app.py +30 -30
app.py CHANGED
@@ -30,16 +30,16 @@ for name in names:
30
  continue
31
  subprocess.run(shlex.split(command), cwd='ControlNet/annotator/ckpts/')
32
 
33
- from gradio_canny2image import create_demo as create_demo_canny
34
- from gradio_depth2image import create_demo as create_demo_depth
35
- from gradio_fake_scribble2image import create_demo as create_demo_fake_scribble
36
- from gradio_hed2image import create_demo as create_demo_hed
37
- from gradio_hough2image import create_demo as create_demo_hough
38
- from gradio_normal2image import create_demo as create_demo_normal
39
- from gradio_pose2image import create_demo as create_demo_pose
40
- from gradio_scribble2image import create_demo as create_demo_scribble
41
- from gradio_scribble2image_interactive import \
42
- create_demo as create_demo_scribble_interactive
43
  from gradio_seg2image import create_demo as create_demo_seg
44
  from model import Model
45
 
@@ -62,27 +62,27 @@ model = Model()
62
  with gr.Blocks(css='style.css') as demo:
63
  gr.Markdown(DESCRIPTION)
64
  with gr.Tabs():
65
- with gr.TabItem('Canny'):
66
- create_demo_canny(model.process_canny, max_images=MAX_IMAGES)
67
- with gr.TabItem('Hough'):
68
- create_demo_hough(model.process_hough, max_images=MAX_IMAGES)
69
- with gr.TabItem('HED'):
70
- create_demo_hed(model.process_hed, max_images=MAX_IMAGES)
71
- with gr.TabItem('Scribble'):
72
- create_demo_scribble(model.process_scribble, max_images=MAX_IMAGES)
73
- with gr.TabItem('Scribble Interactive'):
74
- create_demo_scribble_interactive(
75
- model.process_scribble_interactive, max_images=MAX_IMAGES)
76
- with gr.TabItem('Fake Scribble'):
77
- create_demo_fake_scribble(model.process_fake_scribble,
78
- max_images=MAX_IMAGES)
79
- with gr.TabItem('Pose'):
80
- create_demo_pose(model.process_pose, max_images=MAX_IMAGES)
81
  with gr.TabItem('Segmentation'):
82
  create_demo_seg(model.process_seg, max_images=MAX_IMAGES)
83
- with gr.TabItem('Depth'):
84
- create_demo_depth(model.process_depth, max_images=MAX_IMAGES)
85
- with gr.TabItem('Normal map'):
86
- create_demo_normal(model.process_normal, max_images=MAX_IMAGES)
87
 
88
  demo.queue(api_open=False).launch()
 
30
  continue
31
  subprocess.run(shlex.split(command), cwd='ControlNet/annotator/ckpts/')
32
 
33
+ #from gradio_canny2image import create_demo as create_demo_canny
34
+ #from gradio_depth2image import create_demo as create_demo_depth
35
+ #from gradio_fake_scribble2image import create_demo as create_demo_fake_scribble
36
+ #from gradio_hed2image import create_demo as create_demo_hed
37
+ #from gradio_hough2image import create_demo as create_demo_hough
38
+ #from gradio_normal2image import create_demo as create_demo_normal
39
+ #from gradio_pose2image import create_demo as create_demo_pose
40
+ #from gradio_scribble2image import create_demo as create_demo_scribble
41
+ #from gradio_scribble2image_interactive import \
42
+ # create_demo as create_demo_scribble_interactive
43
  from gradio_seg2image import create_demo as create_demo_seg
44
  from model import Model
45
 
 
62
  with gr.Blocks(css='style.css') as demo:
63
  gr.Markdown(DESCRIPTION)
64
  with gr.Tabs():
65
+ #with gr.TabItem('Canny'):
66
+ # create_demo_canny(model.process_canny, max_images=MAX_IMAGES)
67
+ #with gr.TabItem('Hough'):
68
+ # create_demo_hough(model.process_hough, max_images=MAX_IMAGES)
69
+ #with gr.TabItem('HED'):
70
+ # create_demo_hed(model.process_hed, max_images=MAX_IMAGES)
71
+ #with gr.TabItem('Scribble'):
72
+ # create_demo_scribble(model.process_scribble, max_images=MAX_IMAGES)
73
+ #with gr.TabItem('Scribble Interactive'):
74
+ # create_demo_scribble_interactive(
75
+ # model.process_scribble_interactive, max_images=MAX_IMAGES)
76
+ #with gr.TabItem('Fake Scribble'):
77
+ # create_demo_fake_scribble(model.process_fake_scribble,
78
+ # max_images=MAX_IMAGES)
79
+ #with gr.TabItem('Pose'):
80
+ # create_demo_pose(model.process_pose, max_images=MAX_IMAGES)
81
  with gr.TabItem('Segmentation'):
82
  create_demo_seg(model.process_seg, max_images=MAX_IMAGES)
83
+ #with gr.TabItem('Depth'):
84
+ # create_demo_depth(model.process_depth, max_images=MAX_IMAGES)
85
+ #with gr.TabItem('Normal map'):
86
+ # create_demo_normal(model.process_normal, max_images=MAX_IMAGES)
87
 
88
  demo.queue(api_open=False).launch()