Update app.py

app.py

@@ -9,6 +9,7 @@ from datasets import load_dataset
 import pandas as pd
 import requests
 from io import BytesIO
+import spaces
 
 
 # download model and dataset
@@ -26,13 +27,15 @@ def read_image_from_url(url):
     response = requests.get(url)
     img = Image.open(BytesIO(response.content)).convert("RGB")
     return img
-    
+
+@spaces.GPU
 def extract_features_siglip(image):
     with torch.no_grad():
         inputs = processor(images=image, return_tensors="pt").to(device)
         image_features = model.get_image_features(**inputs)
     return image_features
 
+
 def infer(input_image):
   input_features = extract_features_siglip(input_image["composite"].convert("RGB"))
   input_features = input_features.detach().cpu().numpy()