diff --git a/apps/whisper/Dockerfile b/apps/whisper/Dockerfile index 5ae7b49..22aec7e 100644 --- a/apps/whisper/Dockerfile +++ b/apps/whisper/Dockerfile @@ -12,9 +12,6 @@ apt-get update \ python3-dev \ python3-pip \ \ -# && pip3 uninstall torch \ -# && pip3 cache purge \ -# && pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 \ && pip3 install --no-cache-dir -U \ setuptools \ wheel \ @@ -31,4 +28,4 @@ COPY ./apps/whisper/entrypoint.sh /entrypoint.sh ENV LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu EXPOSE 10300 -ENTRYPOINT ["/entrypoint.sh"] +ENTRYPOINT ["bash", "/entrypoint.sh"] diff --git a/apps/whisper/entrypoint.sh b/apps/whisper/entrypoint.sh index 794a01e..fac9b74 100755 --- a/apps/whisper/entrypoint.sh +++ b/apps/whisper/entrypoint.sh @@ -1,10 +1,6 @@ #!/usr/bin/env bash -#shellcheck disable=SC2086 - -exec \ - python3 -m wyoming_faster_whisper \ +python3 -m wyoming_faster_whisper \ --uri 'tcp://0.0.0.0:10300' \ --data-dir /data \ + --device cuda \ --download-dir /data "$@" - --download-dir /data "$@" \ - --device cuda