Spaces:
Sleeping
Sleeping
fix: typo in small checkpoint name
#3
by
mikhail-shevtsov
- opened
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 == "
|
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)
|