From 96de086509163b8c8aaf5373dffe8fc19b1e5d70 Mon Sep 17 00:00:00 2001
From: Songting <112609742+Plachtaa@users.noreply.github.com>
Date: Thu, 16 Nov 2023 21:08:30 +0800
Subject: [PATCH] revert to whisper large v2

---
 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 dfd496d..c82c8ce 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, n_mels=128).to(model.device)
+    mel = whisper.log_mel_spectrogram(audio).to(model.device)
 
     # detect the spoken language
     _, probs = model.detect_language(mel)