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