OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-16 19:31:52 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #13376 from Thaniel94/add-whisper-language-constraint

Browse Source 
feat: Added WHISPER_LANGUAGE env variable
This commit is contained in:
Tim Jaeryang Baek 2025-05-02 03:08:00 -07:00 committed by GitHub
commit 7b014e44ee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend/open_webui/config.py
View File

@ -2604,6 +2604,7 @@ WHISPER_VAD_FILTER = PersistentConfig(
os.getenv("WHISPER_VAD_FILTER", "False").lower() == "true", os.getenv("WHISPER_VAD_FILTER", "False").lower() == "true",
) )
WHISPER_LANGUAGE = os.getenv("WHISPER_LANGUAGE", "").lower() or None
# Add Deepgram configuration # Add Deepgram configuration
DEEPGRAM_API_KEY = PersistentConfig( DEEPGRAM_API_KEY = PersistentConfig(

1
backend/open_webui/main.py
View File

@ -173,6 +173,7 @@ from open_webui.config import (
WEB_LOADER_ENGINE, WEB_LOADER_ENGINE,
WHISPER_MODEL, WHISPER_MODEL,
WHISPER_VAD_FILTER, WHISPER_VAD_FILTER,
WHISPER_LANGUAGE,
DEEPGRAM_API_KEY, DEEPGRAM_API_KEY,
WHISPER_MODEL_AUTO_UPDATE, WHISPER_MODEL_AUTO_UPDATE,
WHISPER_MODEL_DIR, WHISPER_MODEL_DIR,

2
backend/open_webui/routers/audio.py
View File

@ -33,6 +33,7 @@ from open_webui.config import (
WHISPER_MODEL_AUTO_UPDATE, WHISPER_MODEL_AUTO_UPDATE,
WHISPER_MODEL_DIR, WHISPER_MODEL_DIR,
CACHE_DIR, CACHE_DIR,
WHISPER_LANGUAGE
) )
from open_webui.constants import ERROR_MESSAGES from open_webui.constants import ERROR_MESSAGES
@ -508,6 +509,7 @@ def transcribe(request: Request, file_path):
file_path, file_path,
beam_size=5, beam_size=5,
vad_filter=request.app.state.config.WHISPER_VAD_FILTER, vad_filter=request.app.state.config.WHISPER_VAD_FILTER,
language=WHISPER_LANGUAGE
) )
log.info( log.info(
"Detected language '%s' with probability %f" "Detected language '%s' with probability %f"