my-test-model / download_model.py
b0HIC4's picture
Add pretrained model weights and tokenizer
225006c
raw
history blame contribute delete
454 Bytes
from transformers import AutoTokenizer, AutoModelForSequenceClassification
# Load tokenizer & model
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
# Save both into the current folder
tokenizer.save_pretrained(".")
model.save_pretrained(".")
print("✅ Saved tokenizer and model files locally.")