ComfyUI-Demo / app.py
kadirnar's picture
Update app.py
f5da000
raw
history blame
1.98 kB
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.launch(debug=True, enable_queue=True)
if __name__ == "__main__":
main()