sdp / app.py
xhxhkxh's picture
Update app.py
7599fda
raw
history blame
277 Bytes
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'
)