File size: 282 Bytes
8205a80 |
1 2 3 4 5 6 7 8 9 |
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
|