Piyazon commited on
Commit
07b85e8
·
1 Parent(s): b67ab05
Files changed (3) hide show
  1. Dockerfile +2 -1
  2. README.md +6 -2
  3. start.sh +6 -1
Dockerfile CHANGED
@@ -45,8 +45,9 @@ USER user
45
  ENV HF_MODEL_REPO="piyazon/whisper_uyghur_pt" \
46
  HF_MODEL_FILENAMES="uyghur_whisper_tiny.pt" \
47
  WHISPER_MODEL_FILENAME="uyghur_whisper_tiny.pt" \
48
- WHISPER_LANGUAGE="ug" \
49
  WHISPER_BACKEND="whisper" \
 
50
  PORT="7860"
51
 
52
  EXPOSE 7860
 
45
  ENV HF_MODEL_REPO="piyazon/whisper_uyghur_pt" \
46
  HF_MODEL_FILENAMES="uyghur_whisper_tiny.pt" \
47
  WHISPER_MODEL_FILENAME="uyghur_whisper_tiny.pt" \
48
+ WHISPER_LANGUAGE="uz" \
49
  WHISPER_BACKEND="whisper" \
50
+ WHISPER_BACKEND_POLICY="localagreement" \
51
  PORT="7860"
52
 
53
  EXPOSE 7860
README.md CHANGED
@@ -24,12 +24,16 @@ Set `HF_TOKEN` as a Space secret with read access to the private model repo.
24
  - Hugging Face model repo: `piyazon/whisper_uyghur_pt`
25
  - Downloaded model files: `uyghur_whisper_tiny.pt`
26
  - Started model: `uyghur_whisper_tiny.pt`
 
 
 
27
  - Port: `7860`
28
 
29
  ## Runtime configuration
30
 
31
  - `WHISPER_MODEL_FILENAME=uyghur_whisper_base.pt` starts the base model instead.
32
  - `WHISPER_MODEL_PATH=/path/to/model.pt` skips Hugging Face download.
33
- - `WHISPER_LANGUAGE=auto` can be used if the fork rejects `ug`.
 
34
  - `WHISPERLIVEKIT_ARGS` appends extra CLI arguments, for example
35
- `--backend-policy localagreement`.
 
24
  - Hugging Face model repo: `piyazon/whisper_uyghur_pt`
25
  - Downloaded model files: `uyghur_whisper_tiny.pt`
26
  - Started model: `uyghur_whisper_tiny.pt`
27
+ - Language: `uz`
28
+ - Backend: `whisper`
29
+ - Backend policy: `localagreement`
30
  - Port: `7860`
31
 
32
  ## Runtime configuration
33
 
34
  - `WHISPER_MODEL_FILENAME=uyghur_whisper_base.pt` starts the base model instead.
35
  - `WHISPER_MODEL_PATH=/path/to/model.pt` skips Hugging Face download.
36
+ - `WHISPER_LANGUAGE=ug` can be used if you want to force Uyghur instead of Uzbek.
37
+ - `WHISPER_BACKEND_POLICY=simulstreaming` switches back to the WLK default streaming policy.
38
  - `WHISPERLIVEKIT_ARGS` appends extra CLI arguments, for example
39
+ `--pcm-input`.
start.sh CHANGED
@@ -3,8 +3,9 @@ set -euo pipefail
3
 
4
  PORT="${PORT:-7860}"
5
  MODEL_PATH="${WHISPER_MODEL_PATH:-}"
6
- WHISPER_LANGUAGE="${WHISPER_LANGUAGE:-ug}"
7
  WHISPER_BACKEND="${WHISPER_BACKEND:-whisper}"
 
8
  WHISPERLIVEKIT_ARGS="${WHISPERLIVEKIT_ARGS:-}"
9
 
10
  if [ -z "${MODEL_PATH}" ]; then
@@ -39,6 +40,10 @@ if [ -n "${WHISPER_BACKEND}" ] && grep -q -- "--backend" "${HELP_FILE}"; then
39
  ARGS+=(--backend "${WHISPER_BACKEND}")
40
  fi
41
 
 
 
 
 
42
  if [ -n "${WHISPERLIVEKIT_ARGS}" ]; then
43
  read -r -a EXTRA_ARGS <<<"${WHISPERLIVEKIT_ARGS}"
44
  ARGS+=("${EXTRA_ARGS[@]}")
 
3
 
4
  PORT="${PORT:-7860}"
5
  MODEL_PATH="${WHISPER_MODEL_PATH:-}"
6
+ WHISPER_LANGUAGE="${WHISPER_LANGUAGE:-uz}"
7
  WHISPER_BACKEND="${WHISPER_BACKEND:-whisper}"
8
+ WHISPER_BACKEND_POLICY="${WHISPER_BACKEND_POLICY:-localagreement}"
9
  WHISPERLIVEKIT_ARGS="${WHISPERLIVEKIT_ARGS:-}"
10
 
11
  if [ -z "${MODEL_PATH}" ]; then
 
40
  ARGS+=(--backend "${WHISPER_BACKEND}")
41
  fi
42
 
43
+ if [ -n "${WHISPER_BACKEND_POLICY}" ] && grep -q -- "--backend-policy" "${HELP_FILE}"; then
44
+ ARGS+=(--backend-policy "${WHISPER_BACKEND_POLICY}")
45
+ fi
46
+
47
  if [ -n "${WHISPERLIVEKIT_ARGS}" ]; then
48
  read -r -a EXTRA_ARGS <<<"${WHISPERLIVEKIT_ARGS}"
49
  ARGS+=("${EXTRA_ARGS[@]}")