update

models/vsa_model.py

@@ -6,7 +6,7 @@
 # https://github.com/IDEA-Research/GroundingDINO
 # https://github.com/InternLM/MindSearch
 # --------------------------------------------------------
-
+import spaces
 import os
 import copy
 
@@ -303,6 +303,7 @@ class VisionSearchAssistant:
         )
         self.use_correlate = True
     
+    @spaces.GPU
     def __call__(
         self,
         image: Union[str, Image.Image, np.ndarray],