add decorator

app.py

@@ -7,6 +7,7 @@ import torchvision.transforms as T
 import torch.nn.functional as F
 from lposs import lposs, lposs_plus
 import torch
+import spaces
 
 device = "cpu"
 if torch.cuda.is_available():
@@ -21,6 +22,7 @@ maskclip = MaskClip().to(device)
 dino = DINO().to(device)
 to_torch_tensor = T.Compose([T.Resize(size=448, max_size=2048), T.ToTensor()])
 
+@spaces.GPU
 def segment_image(img: PIL.Image.Image, classnames: str, use_lposs_plus: bool | None) -> tuple[np.ndarray | PIL.Image.Image | str, list[tuple[np.ndarray | tuple[int, int, int, int], str]]]:
     img_tensor = to_torch_tensor(PIL.Image.fromarray(img)).unsqueeze(0).to(device)
     classnames = [c.strip() for c in classnames.split(",")]