Spaces:
Configuration error
Configuration error
| import gradio as gr | |
| from diffusion_webui.helpers import ( | |
| CodeformerUpscalerGenerator, | |
| StableDiffusionControlInpaintNetDepthGenerator, | |
| StableDiffusionControlNetCannyGenerator, | |
| StableDiffusionControlNetDepthGenerator, | |
| StableDiffusionControlNetHEDGenerator, | |
| StableDiffusionControlNetInpaintCannyGenerator, | |
| StableDiffusionControlNetInpaintHedGenerator, | |
| StableDiffusionControlNetInpaintMlsdGenerator, | |
| StableDiffusionControlNetInpaintPoseGenerator, | |
| StableDiffusionControlNetInpaintScribbleGenerator, | |
| StableDiffusionControlNetInpaintSegGenerator, | |
| StableDiffusionControlNetMLSDGenerator, | |
| StableDiffusionControlNetPoseGenerator, | |
| StableDiffusionControlNetScribbleGenerator, | |
| StableDiffusionControlNetSegGenerator, | |
| StableDiffusionImage2ImageGenerator, | |
| StableDiffusionInpaintGenerator, | |
| StableDiffusionText2ImageGenerator, | |
| ) | |
| 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"): | |
| 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__": | |
| main() | |