diff --git a/STT/whisper_stt_handler.py b/STT/whisper_stt_handler.py index 30d93076b79f3dd6440609140674a42c7780435f..06cf613b01156aadcf0ccee9533c0d7aff039930 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 d2a56e13242c3d067064ba3d4311f89a31a278a1..5dc700bf24e2320d0065ab6db40c0adbcf4782b5 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