Ihor commited on
Commit
ba57149
1 Parent(s): 6c8daa1

update model name in the code

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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="facebook/bart-large-mnli")
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('facebook/bart-large-mnli')
76
- tokenizer = AutoTokenizer.from_pretrained('facebook/bart-large-mnli')
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}.'