diff --git a/meetus/workflow.py b/meetus/workflow.py index 17e6038..902eec8 100644 --- a/meetus/workflow.py +++ b/meetus/workflow.py @@ -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 diff --git a/process_meeting.py b/process_meeting.py index 0dd0446..1b2df72 100644 --- a/process_meeting.py +++ b/process_meeting.py @@ -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(