|
|
@@ -19,6 +19,7 @@ VAD_ENABLED = env_bool("WHISPER_VAD", True)
|
|
|
MODEL_SIZE = str(os.getenv("MODEL_SIZE", "small").lower())
|
|
|
COMPUTE_TYPE = os.getenv("WHISPER_COMPUTE_TYPE", "float32").strip().lower()
|
|
|
BEAM_SIZE = int(os.getenv("WHISPER_BEAM_SIZE", "1"))
|
|
|
+HOTWORDS = os.getenv("WHISPER_HOTWORDS", "Satoko").strip()
|
|
|
if BEAM_SIZE < 1:
|
|
|
raise ValueError("WHISPER_BEAM_SIZE doit être supérieur ou égal à 1")
|
|
|
MIN_AUDIO_SEC = float(os.getenv("MIN_AUDIO_SEC", "2.0"))
|
|
|
@@ -55,6 +56,7 @@ def transcribe_audio(audio):
|
|
|
segments, info = model.transcribe(
|
|
|
audio, language="fr", vad_filter=VAD_ENABLED,
|
|
|
beam_size=BEAM_SIZE, condition_on_previous_text=False,
|
|
|
+ hotwords=HOTWORDS or None,
|
|
|
)
|
|
|
return " ".join(seg.text for seg in segments).strip(), info.language
|
|
|
|