Przeglądaj źródła

Ajouter un vocabulaire Whisper configurable avec Satoko par defaut

Andy Esnard 11 godzin temu
rodzic
commit
592ae00327
1 zmienionych plików z 2 dodań i 0 usunięć
  1. 2 0
      app.py

+ 2 - 0
app.py

@@ -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