kadirnar commited on
Commit
02e289c
1 Parent(s): 216e3b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -5
app.py CHANGED
@@ -1,9 +1,16 @@
1
  import gradio as gr
2
 
3
  from diffusion_webui.helpers import (
 
4
  StableDiffusionControlNetCannyGenerator,
5
  StableDiffusionControlNetDepthGenerator,
6
  StableDiffusionControlNetHEDGenerator,
 
 
 
 
 
 
7
  StableDiffusionControlNetMLSDGenerator,
8
  StableDiffusionControlNetPoseGenerator,
9
  StableDiffusionControlNetScribbleGenerator,
@@ -11,7 +18,6 @@ from diffusion_webui.helpers import (
11
  StableDiffusionImage2ImageGenerator,
12
  StableDiffusionInpaintGenerator,
13
  StableDiffusionText2ImageGenerator,
14
- StableDiffusionControlInpaintNetCannyGenerator,
15
  )
16
 
17
 
@@ -42,11 +48,23 @@ def main():
42
  with gr.Tab("Seg"):
43
  StableDiffusionControlNetSegGenerator.app()
44
  with gr.Tab("ControlNet Inpaint"):
45
- StableDiffusionControlInpaintNetCannyGenerator.app()
46
-
47
- app.queue(concurrency_count=1)
 
 
 
 
 
 
 
 
 
 
 
 
48
  app.launch(debug=True, enable_queue=True)
49
 
50
 
51
  if __name__ == "__main__":
52
- main()
 
1
  import gradio as gr
2
 
3
  from diffusion_webui.helpers import (
4
+ StableDiffusionControlInpaintNetDepthGenerator,
5
  StableDiffusionControlNetCannyGenerator,
6
  StableDiffusionControlNetDepthGenerator,
7
  StableDiffusionControlNetHEDGenerator,
8
+ StableDiffusionControlNetInpaintCannyGenerator,
9
+ StableDiffusionControlNetInpaintHedGenerator,
10
+ StableDiffusionControlNetInpaintMlsdGenerator,
11
+ StableDiffusionControlNetInpaintPoseGenerator,
12
+ StableDiffusionControlNetInpaintScribbleGenerator,
13
+ StableDiffusionControlNetInpaintSegGenerator,
14
  StableDiffusionControlNetMLSDGenerator,
15
  StableDiffusionControlNetPoseGenerator,
16
  StableDiffusionControlNetScribbleGenerator,
 
18
  StableDiffusionImage2ImageGenerator,
19
  StableDiffusionInpaintGenerator,
20
  StableDiffusionText2ImageGenerator,
 
21
  )
22
 
23
 
 
48
  with gr.Tab("Seg"):
49
  StableDiffusionControlNetSegGenerator.app()
50
  with gr.Tab("ControlNet Inpaint"):
51
+ with gr.Tab("Canny"):
52
+ StableDiffusionControlNetInpaintCannyGenerator.app()
53
+ with gr.Tab("Depth"):
54
+ StableDiffusionControlInpaintNetDepthGenerator.app()
55
+ with gr.Tab("HED"):
56
+ StableDiffusionControlNetInpaintHedGenerator.app()
57
+ with gr.Tab("MLSD"):
58
+ StableDiffusionControlNetInpaintMlsdGenerator.app()
59
+ with gr.Tab("Pose"):
60
+ StableDiffusionControlNetInpaintPoseGenerator.app()
61
+ with gr.Tab("Scribble"):
62
+ StableDiffusionControlNetInpaintScribbleGenerator.app()
63
+ with gr.Tab("Seg"):
64
+ StableDiffusionControlNetInpaintSegGenerator.app()
65
+
66
  app.launch(debug=True, enable_queue=True)
67
 
68
 
69
  if __name__ == "__main__":
70
+ main()