Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 from diffusers import StableDiffusionPipeline
@@ -35,6 +36,8 @@ image_gen = image_gen.to(device)
 # Load the image-to-text pipeline
 caption_image = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large", device=device)
 
+
+@spaces.GPU(enable_queue=True)
 def generate_image_from_caption(image, num_inference_steps=50, guidance_scale=7.5):
     # Generate the caption
     caption = caption_image(image)[0]['generated_text']