Yayya / app.py
lucasfelezdev's picture
Update app.py
1806551
import gradio as gr
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()