fffiloni commited on
Commit
b308d40
1 Parent(s): a4918b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -125,13 +125,13 @@ def sam_process(input_image, checkpoint, tracking_points, trackings_input_label)
125
  sam2_checkpoint = "./checkpoints/sam2_hiera_tiny.pt"
126
  model_cfg = "sam2_hiera_t.yaml"
127
  elif checkpoint == "samll":
128
- sam2_checkpoint = "./checkpoints/sam2_hiera_tiny.pt"
129
  model_cfg = "sam2_hiera_s.yaml"
130
  elif checkpoint == "base-plus":
131
- sam2_checkpoint = "./checkpoints/sam2_hiera_tiny.pt"
132
  model_cfg = "sam2_hiera_b+.yaml"
133
  elif checkpoint == "large":
134
- sam2_checkpoint = "./checkpoints/sam2_hiera_tiny.pt"
135
  model_cfg = "sam2_hiera_l.yaml"
136
 
137
  sam2_model = build_sam2(model_cfg, sam2_checkpoint, device="cuda")
 
125
  sam2_checkpoint = "./checkpoints/sam2_hiera_tiny.pt"
126
  model_cfg = "sam2_hiera_t.yaml"
127
  elif checkpoint == "samll":
128
+ sam2_checkpoint = "./checkpoints/sam2_hiera_small.pt"
129
  model_cfg = "sam2_hiera_s.yaml"
130
  elif checkpoint == "base-plus":
131
+ sam2_checkpoint = "./checkpoints/sam2_hiera_base_plus.pt"
132
  model_cfg = "sam2_hiera_b+.yaml"
133
  elif checkpoint == "large":
134
+ sam2_checkpoint = "./checkpoints/sam2_hiera_large.pt"
135
  model_cfg = "sam2_hiera_l.yaml"
136
 
137
  sam2_model = build_sam2(model_cfg, sam2_checkpoint, device="cuda")