Make default transcription whisper tiny

sensein · Aug 29, 2024 · 8442083 · 8442083
1 parent c8a9080
commit 8442083
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/b2aiprep/prepare/prepare.py b/src/b2aiprep/prepare/prepare.py
@@ -128,7 +128,7 @@ def wav_to_features(wav_path: Path):
     features["mfcc"] = extract_mfcc_from_audios([audio])[0]
     features["sample_rate"] = audio.sampling_rate
     features["opensmile"] = extract_opensmile_features_from_audios([audio])[0]
-    speech_to_text_model = HFModel(path_or_uri="openai/whisper-large")
+    speech_to_text_model = HFModel(path_or_uri="openai/whisper-tiny")
     features["transcription"] = transcribe_audios(audios=[audio], model=speech_to_text_model)[0]
     logging.disable(logging.NOTSET)
     _logger.setLevel(logging.INFO)