Macedonian-ASR / download_model.py
cigol123's picture
Create download_model.py
d17a8fc verified
raw
history blame
525 Bytes
from transformers import WhisperProcessor, WhisperForConditionalGeneration
import os
# Ensure the cache directory exists
cache_dir = "./.cache"
os.makedirs(cache_dir, exist_ok=True)
# Download the model and processor
print("Downloading Whisper model and processor...")
processor = WhisperProcessor.from_pretrained("openai/whisper-large-v3", cache_dir=cache_dir)
model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large-v3", cache_dir=cache_dir)
print("✓ Model and processor downloaded successfully!")