Hugging Face's logo

Spaces:
NN-BRD
/
hackathon_depth_segment
Runtime error

App Files Community
jens commited on
Commit
0ef8343
·
1 Parent(s): 542a2ef

updated path to weights

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. sam_vit_b_01ec64(1).pth → sam_vit_b_01ec64.pth +0 -0
app.py CHANGED
@@ -6,7 +6,7 @@ import supervision as sv
6
 
7
  def snap(image, video):
8
  MODEL_TYPE = "vit_b"
9
- checkpoint = "checkpoints/sam_vit_b_01ec64.pth"
10
  sam = sam_model_registry[MODEL_TYPE](checkpoint=checkpoint)
11
  mask_generator = SamAutomaticMaskGenerator(sam)
12
  #mask_generator = SamAutomaticMaskGenerator(sam, points_per_side=50)
 
6
 
7
  def snap(image, video):
8
  MODEL_TYPE = "vit_b"
9
+ checkpoint = "sam_vit_b_01ec64.pth"
10
  sam = sam_model_registry[MODEL_TYPE](checkpoint=checkpoint)
11
  mask_generator = SamAutomaticMaskGenerator(sam)
12
  #mask_generator = SamAutomaticMaskGenerator(sam, points_per_side=50)
sam_vit_b_01ec64(1).pth → sam_vit_b_01ec64.pth RENAMED
File without changes