Ihor commited on
Commit
a5c5e8b
1 Parent(s): c67d6cc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -126,7 +126,7 @@ tags:
126
 
127
  **comprehend-it-multilang-base-base**
128
 
129
- This is an encoder-decoder model based on [mT5-base](google/mt5-base) that was trained on multi-language natural language inference datasets as well as on multiple text classification datasets.
130
 
131
  The model demonstrates a better contextual understanding of text and verbalized label because both inputs are encoded by different parts of a model - encoder and decoder respectively.
132
 
@@ -149,8 +149,8 @@ from liqfit.pipeline import ZeroShotClassificationPipeline
149
  from liqfit.models import T5ForZeroShotClassification
150
  from transformers import T5Tokenizer
151
 
152
- model = T5ForZeroShotClassification.from_pretrained('knowledgator/comprehend-it-multilang-base')
153
- tokenizer = T5Tokenizer.from_pretrained('knowledgator/comprehend-it-multilang-base')
154
  classifier = ZeroShotClassificationPipeline(model=model, tokenizer=tokenizer,
155
  hypothesis_template = '{}', encoder_decoder = True)
156
  ```
 
126
 
127
  **comprehend-it-multilang-base-base**
128
 
129
+ This is an encoder-decoder model based on [mT5-base](https://huggingface.co/google/mt5-base) that was trained on multi-language natural language inference datasets as well as on multiple text classification datasets.
130
 
131
  The model demonstrates a better contextual understanding of text and verbalized label because both inputs are encoded by different parts of a model - encoder and decoder respectively.
132
 
 
149
  from liqfit.models import T5ForZeroShotClassification
150
  from transformers import T5Tokenizer
151
 
152
+ model = T5ForZeroShotClassification.from_pretrained('knowledgator/comprehend_it-multilingual-t5-base')
153
+ tokenizer = T5Tokenizer.from_pretrained('knowledgator/comprehend_it-multilingual-t5-base')
154
  classifier = ZeroShotClassificationPipeline(model=model, tokenizer=tokenizer,
155
  hypothesis_template = '{}', encoder_decoder = True)
156
  ```