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)
|