Hugging Face's logo

Spaces:
ovi054
/
text-guided-mask-for-inpainting
Running

App Files Community
ovi054 commited on
Commit
3f0bb73
·
verified ·
1 Parent(s): c9ed57a

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(