Spaces:
Sleeping
Sleeping
File size: 273 Bytes
eda06d2 8f6f616 6287f15 eda06d2 d6f366b eda06d2 c9f4236 eda06d2 |
1 2 3 4 5 6 7 8 9 10 11 |
import spaces
from diffusers import DiffusionPipeline
pipe = DiffusionPipeline.from_pretrained("tiiuae/falcon-7b-instruct")
pipe.to("cuda")
@spaces.GPU
def generate(prompt):
return pipe(prompt).images
gr.Interface(fn=generate, inputs="text", outputs="image").launch() |