File size: 181 Bytes
f08d3c9 |
1 2 3 4 |
from transformers import AutoModelForSequenceClassification
def load_model():
return AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=2) |
f08d3c9 |
1 2 3 4 |
from transformers import AutoModelForSequenceClassification
def load_model():
return AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=2) |