update model name in the code
Browse files
README.md
CHANGED
|
@@ -39,7 +39,7 @@ The model can be loaded with the `zero-shot-classification` pipeline like so:
|
|
| 39 |
```python
|
| 40 |
from transformers import pipeline
|
| 41 |
classifier = pipeline("zero-shot-classification",
|
| 42 |
-
model="
|
| 43 |
```
|
| 44 |
|
| 45 |
You can then use this pipeline to classify sequences into any of the class names you specify.
|
|
@@ -72,8 +72,8 @@ classifier(sequence_to_classify, candidate_labels, multi_label=True)
|
|
| 72 |
```python
|
| 73 |
# pose sequence as a NLI premise and label as a hypothesis
|
| 74 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 75 |
-
nli_model = AutoModelForSequenceClassification.from_pretrained('
|
| 76 |
-
tokenizer = AutoTokenizer.from_pretrained('
|
| 77 |
|
| 78 |
premise = sequence
|
| 79 |
hypothesis = f'This example is {label}.'
|
|
|
|
| 39 |
```python
|
| 40 |
from transformers import pipeline
|
| 41 |
classifier = pipeline("zero-shot-classification",
|
| 42 |
+
model="knowledgator/comprehend_it-base")
|
| 43 |
```
|
| 44 |
|
| 45 |
You can then use this pipeline to classify sequences into any of the class names you specify.
|
|
|
|
| 72 |
```python
|
| 73 |
# pose sequence as a NLI premise and label as a hypothesis
|
| 74 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 75 |
+
nli_model = AutoModelForSequenceClassification.from_pretrained('knowledgator/comprehend_it-base')
|
| 76 |
+
tokenizer = AutoTokenizer.from_pretrained('knowledgator/comprehend_it-base')
|
| 77 |
|
| 78 |
premise = sequence
|
| 79 |
hypothesis = f'This example is {label}.'
|