facebook
/

bart-large-mnli

Zero-Shot Classification

text-classification

Model card Files Files and versions

lysandre HF Staff commited on Aug 9, 2021

Commit

c626438

·

1 Parent(s): df7df4d

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -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')
 premise = sequence
 hypothesis = f'This example is {label}.'

 ```python
 # pose sequence as a NLI premise and label as a hypothesis
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
+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}.'