# 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")