import gradio as gr from diffusion_webui import ( StableDiffusionControlNetGenerator, StableDiffusionControlNetInpaintGenerator, StableDiffusionImage2ImageGenerator, StableDiffusionInpaintGenerator, StableDiffusionText2ImageGenerator, ) def diffusion_app(): app = gr.Blocks() with app: gr.HTML( """

Stable Diffusion + ControlNet + Inpaint

""" ) gr.HTML( """

Follow me for more! Twitter | Github | Linkedin

""" ) 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=2) app.launch(debug=True, enable_queue=True) if __name__ == "__main__": diffusion_app()