Update README.md
Browse files
README.md
CHANGED
@@ -62,8 +62,8 @@ classifier(sequence_to_classify, candidate_labels, multi_class=True)
|
|
62 |
```python
|
63 |
# pose sequence as a NLI premise and label as a hypothesis
|
64 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
65 |
-
nli_model = AutoModelForSequenceClassification.from_pretrained('
|
66 |
-
tokenizer = AutoTokenizer.from_pretrained('
|
67 |
|
68 |
premise = sequence
|
69 |
hypothesis = f'This example is {label}.'
|
|
|
62 |
```python
|
63 |
# pose sequence as a NLI premise and label as a hypothesis
|
64 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
65 |
+
nli_model = AutoModelForSequenceClassification.from_pretrained('facebook/bart-large-mnli')
|
66 |
+
tokenizer = AutoTokenizer.from_pretrained('facebook/bart-large-mnli')
|
67 |
|
68 |
premise = sequence
|
69 |
hypothesis = f'This example is {label}.'
|