revert to whisper large v2

2023-11-16 21:08:30 +08:00
parent bcce0787b3
commit 96de086509
1 changed files with 1 additions and 1 deletions
@@ -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)