added beam search for short audio annotations

Ikaros-521 · May 23, 2023 · 34c5d91 · 34c5d91
1 parent bf560f2
commit 34c5d91
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/short_audio_transcribe.py b/scripts/short_audio_transcribe.py
@@ -23,7 +23,7 @@ def transcribe_one(audio_path):
     print(f"Detected language: {max(probs, key=probs.get)}")
     lang = max(probs, key=probs.get)
     # decode the audio
-    options = whisper.DecodingOptions()
+    options = whisper.DecodingOptions(beam_size=5, best_of=5)
     result = whisper.decode(model, mel, options)
 
     # print the recognized text