fix: typo in small checkpoint name

#3
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -216,7 +216,7 @@ def load_model(checkpoint):
216
  sam2_checkpoint = "./checkpoints/sam2_hiera_tiny.pt"
217
  model_cfg = "sam2_hiera_t.yaml"
218
  return [sam2_checkpoint, model_cfg]
219
- elif checkpoint == "samll":
220
  sam2_checkpoint = "./checkpoints/sam2_hiera_small.pt"
221
  model_cfg = "sam2_hiera_s.yaml"
222
  return [sam2_checkpoint, model_cfg]
@@ -657,4 +657,4 @@ with gr.Blocks(css=css) as demo:
657
  outputs = [output_propagated, output_video, working_frame, available_frames_to_check, reset_prpgt_brn, mask_final_output]
658
  )
659
 
660
- demo.launch(show_api=False, show_error=True)
 
216
  sam2_checkpoint = "./checkpoints/sam2_hiera_tiny.pt"
217
  model_cfg = "sam2_hiera_t.yaml"
218
  return [sam2_checkpoint, model_cfg]
219
+ elif checkpoint == "small":
220
  sam2_checkpoint = "./checkpoints/sam2_hiera_small.pt"
221
  model_cfg = "sam2_hiera_s.yaml"
222
  return [sam2_checkpoint, model_cfg]
 
657
  outputs = [output_propagated, output_video, working_frame, available_frames_to_check, reset_prpgt_brn, mask_final_output]
658
  )
659
 
660
+ demo.launch(show_api=False, show_error=True)