Update app.py

app.py

@@ -13,6 +13,7 @@ from diffusers import StableDiffusionPipeline
 from diffusers import DDIMScheduler
 from transformers import AutoProcessor, BlipForConditionalGeneration
 from share_btn import community_icon_html, loading_icon_html, share_js
+import spaces
 
 # load pipelines
 sd_model_id = "stabilityai/stable-diffusion-2-1-base"
@@ -23,8 +24,7 @@ sem_pipe = SemanticStableDiffusionPipeline.from_pretrained(sd_model_id, torch_dt
 blip_processor = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
 blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base",torch_dtype=torch.float16).to(device)
 
-
-
+@spaces.GPU
 ## IMAGE CPATIONING ##
 def caption_image(input_image):
     inputs = blip_processor(images=input_image, return_tensors="pt").to(device, torch.float16)