From f09cf64f5a296becf139a4300e652b60375857f6 Mon Sep 17 00:00:00 2001
From: andimarafioti <andimarafioti@gmail.com>
Date: Fri, 30 Aug 2024 09:45:49 +0000
Subject: [PATCH] pass language as a parameter to avoid multi-language
detection
---
arguments_classes/whisper_stt_arguments.py | 15 +++++++++++----
1 file changed, 11 insertions(+), 4 deletions(-)
diff --git a/arguments_classes/whisper_stt_arguments.py b/arguments_classes/whisper_stt_arguments.py
index bed382d..ec2f400 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
--
GitLab