| # prepare_model.py | |
| from transformers import AutoModelForSequenceClassification, AutoTokenizer | |
| MODEL = "distilbert-base-uncased-finetuned-sst-2-english" | |
| m = AutoModelForSequenceClassification.from_pretrained(MODEL) | |
| t = AutoTokenizer.from_pretrained(MODEL) | |
| m.save_pretrained(".") # saves config.json + weights (bin/safetensors) | |
| t.save_pretrained(".") # saves tokenizer files | |
| print("Saved model + tokenizer to repo root") |