Update README.md
Browse files
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/
|
153 |
-
tokenizer = T5Tokenizer.from_pretrained('knowledgator/
|
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 |
```
|