mirror of
https://github.com/open-webui/open-webui
synced 2025-06-11 00:49:44 +00:00
Added WHISPER_LANGUAGE env variable. If set to a country's ISO2, constrains Whisper's stt to that language. Detects language as normal if unset
This commit is contained in:
parent
852d9dcbe9
commit
1efa708f83
@ -2510,6 +2510,11 @@ WHISPER_VAD_FILTER = PersistentConfig(
|
|||||||
os.getenv("WHISPER_VAD_FILTER", "False").lower() == "true",
|
os.getenv("WHISPER_VAD_FILTER", "False").lower() == "true",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
WHISPER_LANGUAGE = PersistentConfig(
|
||||||
|
"WHISPER_LANGUAGE",
|
||||||
|
"audio.stt.whisper_language",
|
||||||
|
os.getenv("WHISPER_LANGUAGE", None).lower(),
|
||||||
|
)
|
||||||
|
|
||||||
# Add Deepgram configuration
|
# Add Deepgram configuration
|
||||||
DEEPGRAM_API_KEY = PersistentConfig(
|
DEEPGRAM_API_KEY = PersistentConfig(
|
||||||
|
@ -167,6 +167,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,
|
||||||
@ -791,6 +792,7 @@ app.state.config.STT_MODEL = AUDIO_STT_MODEL
|
|||||||
|
|
||||||
app.state.config.WHISPER_MODEL = WHISPER_MODEL
|
app.state.config.WHISPER_MODEL = WHISPER_MODEL
|
||||||
app.state.config.WHISPER_VAD_FILTER = WHISPER_VAD_FILTER
|
app.state.config.WHISPER_VAD_FILTER = WHISPER_VAD_FILTER
|
||||||
|
app.state.config.WHISPER_LANGUAGE = WHISPER_LANGUAGE
|
||||||
app.state.config.DEEPGRAM_API_KEY = DEEPGRAM_API_KEY
|
app.state.config.DEEPGRAM_API_KEY = DEEPGRAM_API_KEY
|
||||||
|
|
||||||
app.state.config.AUDIO_STT_AZURE_API_KEY = AUDIO_STT_AZURE_API_KEY
|
app.state.config.AUDIO_STT_AZURE_API_KEY = AUDIO_STT_AZURE_API_KEY
|
||||||
|
@ -501,6 +501,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=request.app.state.config.WHISPER_LANGUAGE
|
||||||
)
|
)
|
||||||
log.info(
|
log.info(
|
||||||
"Detected language '%s' with probability %f"
|
"Detected language '%s' with probability %f"
|
||||||
|
Loading…
Reference in New Issue
Block a user