Update README.md
Browse files
README.md
CHANGED
@@ -23,6 +23,8 @@ It can be used for `topic classification`, `sentiment analysis` and as a reranke
|
|
23 |
|
24 |
The model was trained on synthetic data and can be used in commercial applications.
|
25 |
|
|
|
|
|
26 |
### How to use:
|
27 |
First of all, you need to install GLiClass library:
|
28 |
```bash
|
@@ -34,8 +36,8 @@ Than you need to initialize a model and a pipeline:
|
|
34 |
from gliclass import GLiClassModel, ZeroShotClassificationPipeline
|
35 |
from transformers import AutoTokenizer
|
36 |
|
37 |
-
model = GLiClassModel.from_pretrained("knowledgator/gliclass-small-v1.0")
|
38 |
-
tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-small-v1.0")
|
39 |
|
40 |
pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')
|
41 |
|
|
|
23 |
|
24 |
The model was trained on synthetic data and can be used in commercial applications.
|
25 |
|
26 |
+
This model wasn't additionally fine-tuned on any dataset except initial (MoritzLaurer/synthetic_zeroshot_mixtral_v0.1).
|
27 |
+
|
28 |
### How to use:
|
29 |
First of all, you need to install GLiClass library:
|
30 |
```bash
|
|
|
36 |
from gliclass import GLiClassModel, ZeroShotClassificationPipeline
|
37 |
from transformers import AutoTokenizer
|
38 |
|
39 |
+
model = GLiClassModel.from_pretrained("knowledgator/gliclass-small-v1.0-init")
|
40 |
+
tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-small-v1.0-init")
|
41 |
|
42 |
pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')
|
43 |
|