Merge pull request #1 from Rumeysakeskin/Rumeysakeskin-patch-1

add 'initial_prompt'
m-bain · Jan 26, 2025 · fba0ab0 · fba0ab0
2 parents 36d2622 + 7fd2ef4
commit fba0ab0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/whisperx/asr.py b/whisperx/asr.py
@@ -195,10 +195,15 @@ def transcribe(
         print_progress=False,
         combined_progress=False,
         verbose=False,
+        initial_prompt: Optional[str] = None,
+
     ) -> TranscriptionResult:
         if isinstance(audio, str):
             audio = load_audio(audio)
 
+        if initial_prompt is not None:
+            self.options = replace(self.options, initial_prompt=initial_prompt)
+
         def data(audio, segments):
             for seg in segments:
                 f1 = int(seg['start'] * SAMPLE_RATE)