Spaces:
Runtime error
Runtime error
migrating SAM2 space from T4 to ZERO
Browse files- utils/models.py +1 -2
utils/models.py
CHANGED
@@ -6,9 +6,8 @@ from sam2.build_sam import build_sam2
|
|
6 |
from sam2.sam2_image_predictor import SAM2ImagePredictor
|
7 |
|
8 |
BOX_PROMPT_MODE = "box prompt"
|
9 |
-
POINT_PROMPT_MODE = "point prompt"
|
10 |
MASK_GENERATION_MODE = "mask generation"
|
11 |
-
MODE_NAMES = [BOX_PROMPT_MODE,
|
12 |
|
13 |
CHECKPOINT_NAMES = ["tiny", "small", "base_plus", "large"]
|
14 |
CHECKPOINTS = {
|
|
|
6 |
from sam2.sam2_image_predictor import SAM2ImagePredictor
|
7 |
|
8 |
BOX_PROMPT_MODE = "box prompt"
|
|
|
9 |
MASK_GENERATION_MODE = "mask generation"
|
10 |
+
MODE_NAMES = [BOX_PROMPT_MODE, MASK_GENERATION_MODE]
|
11 |
|
12 |
CHECKPOINT_NAMES = ["tiny", "small", "base_plus", "large"]
|
13 |
CHECKPOINTS = {
|