Spaces:
Running
on
A10G
Running
on
A10G
File size: 2,034 Bytes
2204ef0 1e1a292 f5da000 1e1a292 f5da000 3d45160 f5da000 8bf6bdc f5da000 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import gradio as gr
from diffusion_webui.helpers import (
StableDiffusionControlNetCannyGenerator,
StableDiffusionControlNetDepthGenerator,
StableDiffusionControlNetHEDGenerator,
StableDiffusionControlNetMLSDGenerator,
StableDiffusionControlNetPoseGenerator,
StableDiffusionControlNetScribbleGenerator,
StableDiffusionControlNetSegGenerator,
StableDiffusionImage2ImageGenerator,
StableDiffusionInpaintGenerator,
StableDiffusionText2ImageGenerator,
StableDiffusionControlInpaintNetCannyGenerator,
)
def main():
app = gr.Blocks()
with app:
with gr.Row():
with gr.Column():
with gr.Tab("Text2Img"):
StableDiffusionText2ImageGenerator.app()
with gr.Tab("Img2Img"):
StableDiffusionImage2ImageGenerator.app()
with gr.Tab("Inpaint"):
StableDiffusionInpaintGenerator.app()
with gr.Tab("ControlNet"):
with gr.Tab("Canny"):
StableDiffusionControlNetCannyGenerator.app()
with gr.Tab("Depth"):
StableDiffusionControlNetDepthGenerator.app()
with gr.Tab("HED"):
StableDiffusionControlNetHEDGenerator.app()
with gr.Tab("MLSD"):
StableDiffusionControlNetMLSDGenerator.app()
with gr.Tab("Pose"):
StableDiffusionControlNetPoseGenerator.app()
with gr.Tab("Scribble"):
StableDiffusionControlNetScribbleGenerator.app()
with gr.Tab("Seg"):
StableDiffusionControlNetSegGenerator.app()
with gr.Tab("ControlNet Inpaint"):
StableDiffusionControlInpaintNetCannyGenerator.app()
app.queue(concurrency_count=3)
app.launch(debug=True, enable_queue=True)
if __name__ == "__main__":
main() |