Spaces:
Sleeping
Sleeping
File size: 1,307 Bytes
f6e574f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | #!/usr/bin/env python3
"""Download BERT-tiny model upfront to Model folder."""
from pathlib import Path
from transformers import AutoTokenizer, AutoModel
MODEL_NAME = "prajjwal1/bert-tiny"
MODELS_DIR = Path(__file__).resolve().parent / "Model"
MODEL_PATH = MODELS_DIR / "bert-tiny"
def download_model():
"""Download and save the BERT-tiny model."""
print(f"Downloading model: {MODEL_NAME}")
print(f"Target directory: {MODEL_PATH}")
# Create Model directory
MODELS_DIR.mkdir(parents=True, exist_ok=True)
if MODEL_PATH.exists():
print(f"Model already exists at {MODEL_PATH}")
print("Skipping download.")
return
try:
print("Downloading tokenizer...")
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
print("Downloading model...")
model = AutoModel.from_pretrained(MODEL_NAME)
print(f"Saving model to {MODEL_PATH}...")
tokenizer.save_pretrained(str(MODEL_PATH))
model.save_pretrained(str(MODEL_PATH))
print("✅ Model downloaded and saved successfully!")
print(f"Location: {MODEL_PATH}")
except Exception as e:
print(f"❌ Error downloading model: {e}")
raise
if __name__ == "__main__":
download_model()
|