From f3e86195e14310b0ed69d1598538e496d2eeb2b8 Mon Sep 17 00:00:00 2001
From: Eustache Le Bihan <eulebihan@gmail.com>
Date: Fri, 20 Sep 2024 18:45:02 +0200
Subject: [PATCH] add auto lang signal for llm
---
STT/whisper_stt_handler.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/STT/whisper_stt_handler.py b/STT/whisper_stt_handler.py
index 06cf613..172a01e 100644
--- a/STT/whisper_stt_handler.py
+++ b/STT/whisper_stt_handler.py
@@ -137,4 +137,7 @@ class WhisperSTTHandler(BaseHandler):
console.print(f"[yellow]USER: {pred_text}")
logger.debug(f"Language Code Whisper: {language_code}")
+ if self.language is None:
+ language_code += "-auto"
+
yield (pred_text, language_code)
--
GitLab