Update README.md
This commit is contained in:
parent
df7df4d210
commit
c626438eec
|
@ -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}.'
|
||||
|
|
Loading…
Reference in New Issue