app.py

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()