diff --git a/.gitignore b/.gitignore index ed514b8..eb9b2f2 100644 --- a/.gitignore +++ b/.gitignore @@ -13,5 +13,3 @@ __pycache__/ *.pyc *.pyo -# YOLO model (auto-downloaded on first run) -yolov8n.pt diff --git a/run_all.sh b/run_all.sh index d65843e..db93521 100755 --- a/run_all.sh +++ b/run_all.sh @@ -1,6 +1,11 @@ #!/usr/bin/env bash set -euo pipefail -export PYTHONPATH="/workspace/.site-packages:$PYTHONPATH" +# Link pip deps from EFS vendor if /workspace/.site-packages is missing +if [[ ! -d "${ROOT}/.site-packages" && -d "/opt/vendor/yolov8" ]]; then + ln -sf /opt/vendor/yolov8 "${ROOT}/.site-packages" +fi + +export PYTHONPATH="${ROOT}/.site-packages:$PYTHONPATH" export DISPLAY=:1 # Auto-download YOLO model if not present diff --git a/yolov8n.pt b/yolov8n.pt new file mode 100644 index 0000000..0db4ca4 Binary files /dev/null and b/yolov8n.pt differ