diff --git a/tests/demo/test_pipeline.py b/tests/demo/test_pipeline.py index d8c8142..23ab0fa 100644 --- a/tests/demo/test_pipeline.py +++ b/tests/demo/test_pipeline.py @@ -31,7 +31,7 @@ REPO_ROOT: Final[Path] = Path(__file__).resolve().parents[2] SAMPLE_VIDEO_PATH: Final[Path] = REPO_ROOT / "assets" / "sample.mp4" CHECKPOINT_PATH: Final[Path] = REPO_ROOT / "ckpt" / "ScoNet-20000.pt" CONFIG_PATH: Final[Path] = REPO_ROOT / "configs" / "sconet" / "sconet_scoliosis1k.yaml" -YOLO_MODEL_PATH: Final[Path] = REPO_ROOT / "yolo11n-seg.pt" +YOLO_MODEL_PATH: Final[Path] = REPO_ROOT / "ckpt" / "yolo11n-seg.pt" def _device_for_runtime() -> str: