add languaje en

This commit is contained in:
Mariano Gabriel
2026-06-09 11:08:09 -03:00
parent eb8b1f4f11
commit 8320147f50
2 changed files with 8 additions and 0 deletions

View File

@@ -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

View File

@@ -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(