Karumoon's picture
Update app.py
1ce11e8
raw
history blame
309 Bytes
import gradio as gr
from diffusers import DiffusionPipeline
pipeline = DiffusionPipeline.from_pretrained("anton-l/ddpm-ema-flowers-64")
generator.to("cuda")
def greet(name):
image = generator().images[0]
return image
iface = gr.Interface(fn=greet, inputs="text", outputs="image")
iface.launch()