hm

sam2_mask.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import os
 os.environ["TORCH_CUDNN_SDPA_ENABLED"] = "1"
@@ -106,6 +107,7 @@ def show_masks(image, masks, scores, point_coords=None, box_coords=None, input_l
         plt.close()  # Close the figure to free up memory
     return combined_images, mask_images
 
+@spaces.GPU()
 def sam_process(input_image, tracking_points, trackings_input_label):
     image = Image.open(input_image)
     image = np.array(image.convert("RGB"))