Hugging Face's logo

Spaces:
sahirp
/
text_2_segment_any_vid
Running on Zero

App Files Community
2
er1t0 commited on
Commit
dc2d01b
•
1 Parent(s): aab55b1

dependancies

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -18,7 +18,7 @@ if torch.cuda.get_device_properties(0).major >= 8:
18
  torch.backends.cudnn.allow_tf32 = True
19
 
20
  # Initialize models
21
- sam2_checkpoint = "../checkpoints/sam2_hiera_large.pt"
22
  model_cfg = "sam2_hiera_l.yaml"
23
 
24
  video_predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint)
 
18
  torch.backends.cudnn.allow_tf32 = True
19
 
20
  # Initialize models
21
+ sam2_checkpoint = "./checkpoints/sam2_hiera_large.pt"
22
  model_cfg = "sam2_hiera_l.yaml"
23
 
24
  video_predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint)
requirements.txt CHANGED
@@ -7,4 +7,5 @@ transformers
7
  opencv-python
8
  matplotlib
9
  einops
10
- timm
 
 
7
  opencv-python
8
  matplotlib
9
  einops
10
+ timm
11
+ pytest