Yayya / app.py
lucasfelezdev's picture
Update app.py
88af508
raw
history blame
305 Bytes
import gradio as gr
import diffusers
from diffusers import DiffusionPipeline
pipeline = DiffusionPipeline.from_pretrained("lucasfelezdev/TextToImage")
def greet(text):
image = pipeline("text").images[0]
return image
iface = gr.Interface(fn=greet, inputs="text", outputs="image")
iface.launch()