File size: 386 Bytes
b730de8
1ce11e8
 
cccf69a
 
1ce11e8
dc97217
1ce11e8
b730de8
 
1ce11e8
 
b730de8
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import gradio as gr
from diffusers import DiffusionPipeline

generator = DiffusionPipeline.from_pretrained("anton-l/ddpm-ema-flowers-64")
#generator = DiffusionPipeline.from_pretrained("google/ddpm-celebahq-256")

#generator.to("cuda")


def greet(name):
    image = generator().images[0]
    return image


iface = gr.Interface(fn=greet, inputs="text", outputs="image")
iface.launch()