diff --git a/STT/whisper_stt_handler.py b/STT/whisper_stt_handler.py index e3c99bd2310c97d2eded7ba4bce5427ebe3a3e4a..d669e34eac30656f806f96ad4a6a65a73f11b237 100644 --- a/STT/whisper_stt_handler.py +++ b/STT/whisper_stt_handler.py @@ -40,6 +40,8 @@ class WhisperSTTHandler(BaseHandler): self.torch_dtype = getattr(torch, torch_dtype) self.compile_mode = compile_mode self.gen_kwargs = gen_kwargs + if language == 'None': + language = None self.last_language = language if self.last_language is not None: self.gen_kwargs["language"] = self.last_language diff --git a/arguments_classes/whisper_stt_arguments.py b/arguments_classes/whisper_stt_arguments.py index ec2f400ab7de9dde45fa45c0fa8056aa35053a37..7073e29d4de0433af83d24a042d54a4c6704d1c0 100644 --- a/arguments_classes/whisper_stt_arguments.py +++ b/arguments_classes/whisper_stt_arguments.py @@ -59,8 +59,12 @@ class WhisperSTTHandlerArguments: }, ) language: Optional[str] = field( - default=None, + default='en', metadata={ - "help": "The language for the conversation. Default is None." + "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 + change during the conversation. Default is 'en'.""" }, ) \ No newline at end of file