File size: 309 Bytes
b730de8
1ce11e8
 
 
 
 
 
b730de8
 
1ce11e8
 
b730de8
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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()