import gradio as gr from diffusers import StableDiffusionPipeline def draw(sets): pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-EN-v0.1") images = pipe(sets, guidance_scale=10).images[] return images[0],images[1],images[2],images[3],images[4] iface = gr.Interface(fn=draw, inputs="text", outputs=["image","image","image","image","image"],) iface.launch()