devil2810's picture
Upload 7 files
8205a80 verified
raw
history blame contribute delete
282 Bytes
from transformers import AutoTokenizer, AutoModelForSequenceClassification
def load_model(model_id: str):
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSequenceClassification.from_pretrained(model_id)
model.eval()
return tokenizer, model