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' | |
) | |