load_model.py

import torch
from transformers import AutoModelForSequenceClassification

def load_model(model_path):
    model = AutoModelForSequenceClassification.from_pretrained(model_path)
    model.eval()
    return model