test everything running on gpu

app.py

@@ -333,6 +333,7 @@ def load_img_from_URL(URL):
     init_image = Image.open(BytesIO(response.content)).convert("RGB")
     return init_image
 
+@spaces.GPU
 def embed_img(input_image):
     tokens = pipe.feature_extractor(input_image)
     img_model = pipe.image_encoder.to(torch_device)
@@ -341,7 +342,7 @@ def embed_img(input_image):
         
     return embeds.image_embeds.to(torch_device)
 
-
+@spaces.GPU
 def localimg_2_embed(image_dir):
     embeds = embed_img(load_image(image_dir))
     return embeds