import gradio as gr from diffusion_webui import ( StableDiffusionControlNetGenerator, StableDiffusionControlNetInpaintGenerator, StableDiffusionImage2ImageGenerator, StableDiffusionInpaintGenerator, StableDiffusionText2ImageGenerator, ) def diffusion_app(): app = gr.Blocks() with app: with gr.Row(): with gr.Column(): with gr.Tab(label="Text2Image"): StableDiffusionText2ImageGenerator.app() with gr.Tab(label="Image2Image"): StableDiffusionImage2ImageGenerator.app() with gr.Tab(label="Inpaint"): StableDiffusionInpaintGenerator.app() with gr.Tab(label="Controlnet"): StableDiffusionControlNetGenerator.app() with gr.Tab(label="Controlnet Inpaint"): StableDiffusionControlNetInpaintGenerator.app() app.queue(concurrency_count=1) app.launch(debug=True, enable_queue=True) if __name__ == "__main__": diffusion_app()