sdp / app.py
xhxhkxh's picture
Update app.py
3f314bd
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()