mariano/meetus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add languaje en

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

3
meetus/workflow.py
View File

@@ -34,6 +34,7 @@ class WorkflowConfig:
self.run_whisper = kwargs.get('run_whisper', False) self.run_whisper = kwargs.get('run_whisper', False)
self.whisper_model = kwargs.get('whisper_model', 'medium') self.whisper_model = kwargs.get('whisper_model', 'medium')
self.diarize = kwargs.get('diarize', False) self.diarize = kwargs.get('diarize', False)
self.language = kwargs.get('language')
# Frame extraction # Frame extraction
self.scene_detection = kwargs.get('scene_detection', False) self.scene_detection = kwargs.get('scene_detection', False)
@@ -226,6 +227,8 @@ class ProcessingWorkflow:
] ]
if use_diarize: if use_diarize:
cmd.append("--diarize") cmd.append("--diarize")
if self.config.language:
cmd.extend(["--language", self.config.language])
try: try:
# Set up environment with cuDNN library path for whisperx # Set up environment with cuDNN library path for whisperx

5
process_meeting.py
View File

@@ -77,6 +77,11 @@ Examples:
action='store_true', action='store_true',
help='Use WhisperX with speaker diarization (requires whisperx and HuggingFace token)' 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 # Output options
parser.add_argument( parser.add_argument(