File size: 301 Bytes
7338c67
 
d5e5583
8fa8eb0
7338c67
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import spaces
from diffusers import DiffusionPipeline

pipe = DiffusionPipeline.from_pretrained('ptx0/terminus-xl-velocity-v2')
pipe.to('cuda')

@spaces.GPU
def generate(prompt):
    return pipe(prompt).images

gr.Interface(
    fn=generate,
    inputs=gr.Text(),
    outputs=gr.Gallery(),
).launch()