Hugging Face's logo

Spaces:
jotase
/
florence-sam-masking
Running on Zero

App Files Community
jotase commited on
Commit
63d2846
·
verified ·
1 Parent(s): e69b3f5

Update utils/sam.py

Browse files
Files changed (1) hide show
  1. utils/sam.py +4 -4
utils/sam.py CHANGED
@@ -7,10 +7,10 @@ from PIL import Image
7
  from sam2.build_sam import build_sam2, build_sam2_video_predictor
8
  from sam2.sam2_image_predictor import SAM2ImagePredictor
9
 
10
- SAM_CHECKPOINT = "checkpoints/sam2_hiera_small.pt"
11
- SAM_CONFIG = "sam2_hiera_s.yaml"
12
- # SAM_CHECKPOINT = "checkpoints/sam2_hiera_large.pt"
13
- # SAM_CONFIG = "sam2_hiera_l.yaml"
14
 
15
 
16
  def load_sam_image_model(
 
7
  from sam2.build_sam import build_sam2, build_sam2_video_predictor
8
  from sam2.sam2_image_predictor import SAM2ImagePredictor
9
 
10
+ # SAM_CHECKPOINT = "checkpoints/sam2_hiera_small.pt"
11
+ # SAM_CONFIG = "sam2_hiera_s.yaml"
12
+ SAM_CHECKPOINT = "checkpoints/sam2_hiera_large.pt"
13
+ SAM_CONFIG = "sam2_hiera_l.yaml"
14
 
15
 
16
  def load_sam_image_model(