add languaje en
This commit is contained in:
@@ -34,6 +34,7 @@ class WorkflowConfig:
|
||||
self.run_whisper = kwargs.get('run_whisper', False)
|
||||
self.whisper_model = kwargs.get('whisper_model', 'medium')
|
||||
self.diarize = kwargs.get('diarize', False)
|
||||
self.language = kwargs.get('language')
|
||||
|
||||
# Frame extraction
|
||||
self.scene_detection = kwargs.get('scene_detection', False)
|
||||
@@ -226,6 +227,8 @@ class ProcessingWorkflow:
|
||||
]
|
||||
if use_diarize:
|
||||
cmd.append("--diarize")
|
||||
if self.config.language:
|
||||
cmd.extend(["--language", self.config.language])
|
||||
|
||||
try:
|
||||
# Set up environment with cuDNN library path for whisperx
|
||||
|
||||
@@ -77,6 +77,11 @@ Examples:
|
||||
action='store_true',
|
||||
help='Use WhisperX with speaker diarization (requires whisperx and HuggingFace token)'
|
||||
)
|
||||
parser.add_argument(
|
||||
'--language',
|
||||
help='Language of the audio (e.g. en, es, fr). Default: auto-detect',
|
||||
default=None
|
||||
)
|
||||
|
||||
# Output options
|
||||
parser.add_argument(
|
||||
|
||||
Reference in New Issue
Block a user