PengWeixuanSZU commited on
Commit
9505657
·
verified ·
1 Parent(s): bc670af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -45,6 +45,8 @@ from sam2 import load_model
45
  from sam2.build_sam import build_sam2, build_sam2_video_predictor
46
  from sam2.sam2_image_predictor import SAM2ImagePredictor
47
 
 
 
48
  COLOR_PALETTE = [
49
  (255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 255, 0), (255, 0, 255),
50
  (0, 255, 255), (255, 128, 0), (128, 0, 255), (0, 128, 255), (128, 255, 0)
 
45
  from sam2.build_sam import build_sam2, build_sam2_video_predictor
46
  from sam2.sam2_image_predictor import SAM2ImagePredictor
47
 
48
+ import spaces
49
+
50
  COLOR_PALETTE = [
51
  (255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 255, 0), (255, 0, 255),
52
  (0, 255, 255), (255, 128, 0), (128, 0, 255), (0, 128, 255), (128, 255, 0)