From 05cd65714cc9f75bee3532f276721f463c10575e Mon Sep 17 00:00:00 2001
From: Robert Dunmire III <slackr31337@gmail.com>
Date: Tue, 14 Jan 2025 07:44:03 -0500
Subject: [PATCH] Update README.md and run.sh

---
 .vscode/settings.json |  9 +++++++++
 README.md             | 26 ++++++++++++++++++++++++--
 run.sh                |  4 +---
 3 files changed, 34 insertions(+), 5 deletions(-)
 create mode 100644 .vscode/settings.json

diff --git a/.vscode/settings.json b/.vscode/settings.json
new file mode 100644
index 0000000..cab12f8
--- /dev/null
+++ b/.vscode/settings.json
@@ -0,0 +1,9 @@
+{
+    "cSpell.words": [
+        "cuda",
+        "cudnn",
+        "noninteractive",
+        "slackr",
+        "venv"
+    ]
+}
\ No newline at end of file
diff --git a/README.md b/README.md
index 9b65bee..c64b55b 100644
--- a/README.md
+++ b/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"
 >
diff --git a/run.sh b/run.sh
index 4a5f368..b5940b8 100644
--- a/run.sh
+++ b/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 \
     "$@"