import spaces | |
from diffusers import DiffusionPipeline | |
pipe = DiffusionPipeline.from_pretrained('ptx0/terminus-xl-velocity-v2') | |
pipe.to('cuda') | |
def generate(prompt): | |
return pipe(prompt).images | |
gr.Interface( | |
fn=generate, | |
inputs=gr.Text(), | |
outputs=gr.Gallery(), | |
).launch() | |