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}.'
|