Update florence_sam/utils/florence.py

florence_sam/utils/florence.py

@@ -1,3 +1,4 @@
+import spaces
 import os
 from typing import Union, Any, Tuple, Dict
 from unittest.mock import patch
@@ -23,7 +24,7 @@ def fixed_get_imports(filename: Union[str, os.PathLike]) -> list[str]:
     #imports.remove("flash_attn")
     return imports
 
-
+@spaces.GPU 
 def load_florence_model(
     device: torch.device, checkpoint: str = FLORENCE_CHECKPOINT
 ) -> Tuple[Any, Any]: