diff --git a/scripts/install_helpers.sh b/scripts/install_helpers.sh index ebcc6e0..76a75ff 100644 --- a/scripts/install_helpers.sh +++ b/scripts/install_helpers.sh @@ -3,17 +3,24 @@ get_tf_whl () { BASE_URL=https://github.com/Nachtzuster/BirdNET-Pi/releases/download/v0.1/ + ARCH=$(uname -m) PY_VERSION=$(python3 -c "import sys; print(f'{sys.version_info[0]}{sys.version_info[1]}')") - case "${PY_VERSION}" in - 39) + case "${ARCH}-${PY_VERSION}" in + aarch64-39) WHL=tflite_runtime-2.11.0-cp39-none-linux_aarch64.whl ;; - 310) + aarch64-310) WHL=tflite_runtime-2.11.0-cp310-none-linux_aarch64.whl ;; - 311) + aarch64-311) WHL=tflite_runtime-2.11.0-cp311-cp311-linux_aarch64.whl ;; + x86_64-39) + WHL=tflite_runtime-2.11.0-cp39-cp39-linux_x86_64.whl + ;; + x86_64-311) + WHL=tflite_runtime-2.11.0-cp311-cp311-linux_x86_64.whl + ;; *) echo "No tflite version found" WHL=''