from transformers import AutoModel, AutoTokenizer

MODEL_NAME = "distilbert-base-uncased"

def load_model():
    tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
    model = AutoModel.from_pretrained(MODEL_NAME)
    return tokenizer, model

if __name__ == "__main__":
    tokenizer, model = load_model()
    print("Model loaded successfully")