From 1f102f3c0914f295ff25ba9020682322113d6580 Mon Sep 17 00:00:00 2001
From: Danu Kim <71139863+DogeLord081@users.noreply.github.com>
Date: Mon, 13 Nov 2023 16:56:55 -0500
Subject: [PATCH] Update short_audio_transcribe.py

---
 scripts/short_audio_transcribe.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/short_audio_transcribe.py b/scripts/short_audio_transcribe.py
index c82c8ce..dfd496d 100644
--- a/scripts/short_audio_transcribe.py
+++ b/scripts/short_audio_transcribe.py
@@ -16,7 +16,7 @@ def transcribe_one(audio_path):
     audio = whisper.pad_or_trim(audio)
 
     # make log-Mel spectrogram and move to the same device as the model
-    mel = whisper.log_mel_spectrogram(audio).to(model.device)
+    mel = whisper.log_mel_spectrogram(audio, n_mels=128).to(model.device)
 
     # detect the spoken language
     _, probs = model.detect_language(mel)