Spaces:
Paused
Paused
File size: 3,376 Bytes
50bd6a9 |
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 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
import gradio as gr
from diffusion_webui import (
CodeformerUpscalerGenerator,
StableDiffusionControlInpaintNetDepthGenerator,
StableDiffusionControlNetCannyGenerator,
StableDiffusionControlNetDepthGenerator,
StableDiffusionControlNetHEDGenerator,
StableDiffusionControlNetInpaintCannyGenerator,
StableDiffusionControlNetInpaintHedGenerator,
StableDiffusionControlNetInpaintMlsdGenerator,
StableDiffusionControlNetInpaintPoseGenerator,
StableDiffusionControlNetInpaintScribbleGenerator,
StableDiffusionControlNetInpaintSegGenerator,
StableDiffusionControlNetMLSDGenerator,
StableDiffusionControlNetPoseGenerator,
StableDiffusionControlNetScribbleGenerator,
StableDiffusionControlNetSegGenerator,
StableDiffusionImage2ImageGenerator,
StableDiffusionInpaintGenerator,
StableDiffusionText2ImageGenerator,
StableDiffusionControlNetNormalGenerator,
)
def diffusion_app():
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("Normal"):
StableDiffusionControlNetNormalGenerator.app()
with gr.Tab("Seg"):
StableDiffusionControlNetSegGenerator.app()
with gr.Tab("ControlNet Inpaint"):
with gr.Tab("Canny"):
StableDiffusionControlNetInpaintCannyGenerator.app()
with gr.Tab("Depth"):
StableDiffusionControlInpaintNetDepthGenerator.app()
with gr.Tab("HED"):
StableDiffusionControlNetInpaintHedGenerator.app()
with gr.Tab("MLSD"):
StableDiffusionControlNetInpaintMlsdGenerator.app()
with gr.Tab("Pose"):
StableDiffusionControlNetInpaintPoseGenerator.app()
with gr.Tab("Scribble"):
StableDiffusionControlNetInpaintScribbleGenerator.app()
with gr.Tab("Seg"):
StableDiffusionControlNetInpaintSegGenerator.app()
with gr.Tab("Upscaler"):
CodeformerUpscalerGenerator.app()
app.queue(concurrency_count=2)
app.launch(debug=True, enable_queue=True)
if __name__ == "__main__":
diffusion_app()
|