ThinhLLM/IMDB_Sentiment_Classification_With_Bert

1 2from transformers import AutoTokenizer, AutoModelForSequenceClassification3 4def create_bert():5    model_ckpt = "bert-base-uncased"6    tokenizer = AutoTokenizer.from_pretrained(model_ckpt)7    model = AutoModelForSequenceClassification.from_pretrained(model_ckpt,num_labels=2)8    return model, tokenizer9