huggingface
/
bart-large-mnli
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
Lysandre 2021-08-09 08:25:07 +00:00 committed by huggingface-web
parent df7df4d210
commit c626438eec
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

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