diff --git a/arguments_classes/whisper_stt_arguments.py b/arguments_classes/whisper_stt_arguments.py index bed382dda754da36965b4d86e68a7f8b4d9c322c..ec2f400ab7de9dde45fa45c0fa8056aa35053a37 100644 --- a/arguments_classes/whisper_stt_arguments.py +++ b/arguments_classes/whisper_stt_arguments.py @@ -1,8 +1,15 @@ from dataclasses import dataclass, field +from typing import Optional @dataclass class WhisperSTTHandlerArguments: + language: Optional[str] = field( + default=None, + metadata={ + "help": "The language for the conversation. Default is None." + }, + ) stt_model_name: str = field( default="distil-whisper/distil-large-v3", metadata={ @@ -51,9 +58,9 @@ class WhisperSTTHandlerArguments: "help": "The task to perform, typically 'transcribe' for transcription. Default is 'transcribe'." }, ) - stt_gen_language: str = field( - default="en", + language: Optional[str] = field( + default=None, metadata={ - "help": "The language of the speech to transcribe. Default is 'en' for English." + "help": "The language for the conversation. Default is None." }, - ) + ) \ No newline at end of file