From 4e6055f1a995c666aab2acbb5809408ca0251b40 Mon Sep 17 00:00:00 2001
From: Andres Marafioti <andimarafioti@gmail.com>
Date: Wed, 4 Sep 2024 13:53:47 +0200
Subject: [PATCH] pass auto for auto language detection
---
STT/whisper_stt_handler.py | 2 +-
arguments_classes/whisper_stt_arguments.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/STT/whisper_stt_handler.py b/STT/whisper_stt_handler.py
index 30d9307..06cf613 100644
--- a/STT/whisper_stt_handler.py
+++ b/STT/whisper_stt_handler.py
@@ -40,7 +40,7 @@ class WhisperSTTHandler(BaseHandler):
self.torch_dtype = getattr(torch, torch_dtype)
self.compile_mode = compile_mode
self.gen_kwargs = gen_kwargs
- if language == 'None':
+ if language == 'auto':
language = None
self.last_language = language
if self.last_language is not None:
diff --git a/arguments_classes/whisper_stt_arguments.py b/arguments_classes/whisper_stt_arguments.py
index d2a56e1..5dc700b 100644
--- a/arguments_classes/whisper_stt_arguments.py
+++ b/arguments_classes/whisper_stt_arguments.py
@@ -58,7 +58,7 @@ class WhisperSTTHandlerArguments:
"help": """The language for the conversation.
Choose between 'en' (english), 'fr' (french), 'es' (spanish),
'zh' (chinese), 'ko' (korean), 'ja' (japanese), or 'None'.
- If using None, the language is automatically detected and can
+ If using 'auto', the language is automatically detected and can
change during the conversation. Default is 'en'."""
},
)
\ No newline at end of file
--
GitLab