update whisper

hagak · Nov 30, 2024 · b09f6d3 · b09f6d3
1 parent 490ea72
commit b09f6d3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git 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
@@ -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