from transformers import AutoModelForSequenceClassification

def load_model():
    return AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=2)