sdp / app.py
xhxhkxh's picture
Update app.py
07272f1
raw
history blame
278 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'
)