Update app.py

app.py

@@ -24,6 +24,8 @@ from UniVAD.models.segment_anything import (
     SamPredictor,
 )
 
+import spaces
+
 
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -61,7 +63,8 @@ def preprocess_image(img):
 def update_image(image):
     if image is not None:
         return preprocess_image(image)
-    
+
+@spaces.GPU
 def ad(image_pil, normal_image, box_threshold, text_threshold, text_prompt, background_prompt, cluster_num):
     return process_image(image_pil, normal_image, box_threshold, text_threshold, sam_predictor, grounding_model, univad_model, ram_model, text_prompt, background_prompt, cluster_num, image_size, grounding_processor)