from diffusers import DiffusionPipeline import gradio as gr def gen(text): pipeline = DiffusionPipeline.from_pretrained("PublicPrompts/All-In-One-Pixel-Model") return pipeline(text).images[0] gr.Interface(fn=gen,inputs='text' ,outputs='image' ).launch()