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