diff --git a/README.md b/README.md index bb29a53..7e9ace4 100644 --- a/README.md +++ b/README.md @@ -62,8 +62,8 @@ classifier(sequence_to_classify, candidate_labels, multi_class=True) ```python # pose sequence as a NLI premise and label as a hypothesis from transformers import AutoModelForSequenceClassification, AutoTokenizer -nli_model = AutoModelForSequenceClassification.from_pretrained('joeddav/xlm-roberta-large-xnli') -tokenizer = AutoTokenizer.from_pretrained('joeddav/xlm-roberta-large-xnli') +nli_model = AutoModelForSequenceClassification.from_pretrained('facebook/bart-large-mnli') +tokenizer = AutoTokenizer.from_pretrained('facebook/bart-large-mnli') premise = sequence hypothesis = f'This example is {label}.'