Update README.md and run.sh

.vscode/settings.json

@@ -0,0 +1,9 @@
+{
+    "cSpell.words": [
+        "cuda",
+        "cudnn",
+        "noninteractive",
+        "slackr",
+        "venv"
+    ]
+}

README.md

@@ -9,9 +9,31 @@ https://github.com/rhasspy/wyoming-faster-whisper
 docker pull ghcr.io/slackr31337/wyoming-whisper-gpu:latest
 
 
-# Use environment variable to set model (default: tiny-int8)
+# Use environment variable to set model
 
-> MODEL="Systran/faster-distil-whisper-small.en"
+default: tiny-int8
+
+tiny-int8
+tiny.en
+tiny
+base-int8
+base.en
+base
+small-int8
+small.en
+small
+medium-int8
+medium.en
+medium
+large-v1
+large-v2
+large-v3
+large
+distil-large-v2
+distil-medium.en
+distil-small.en
+
+> MODEL="base-int8"
 >
 > LANGUAGE="en"
 >

run.sh

@@ -1,12 +1,10 @@
 #!/usr/bin/env bash
 
-export LD_LIBRARY_PATH=/app/lib:/app/lib/python3.10/dist-packages/nvidia/cudnn/lib/:$LD_LIBRARY_PATH
-
-source /app/bin/activate
 /app/bin/python3 -m  wyoming_faster_whisper \
     --uri 'tcp://0.0.0.0:10300' \
     --data-dir /data \
     --download-dir /data \
     --model "${MODEL:-tiny-int8}" \
     --language "${LANGUAGE:-en}" \
+    --device cuda \
     "$@"