create folder (#4390)

This commit is contained in:
martinb-bb
2023-03-03 10:03:16 -05:00
committed by GitHub
parent 94e2b33802
commit 6049a0686c

View File

@@ -163,6 +163,8 @@ def transcribe_and_summarize(
# cache directory is ~/.cache/whisper
whisper_cache_dir = os.path.join(os.path.expanduser("~"), ".cache", "whisper")
os.makedirs(whisper_cache_dir, exist_ok=True)
if model_name in ["small", "small.en"]:
model_size = "483.6 MB"
elif model_name in ["tiny", "tiny.en"]: