Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 import numpy as np
@@ -37,6 +38,7 @@ def overlay_masks(image: Image.Image, masks: torch.Tensor) -> Image.Image:
         image = Image.alpha_composite(image, overlay)
     return image
 
+spaces.GPU()
 def segment(image: Image.Image, text: str, threshold: float, mask_threshold: float):
     """
     Perform promptable concept segmentation using SAM3.