Zero-Shot Classification
Transformers
ONNX
Safetensors
English
GLiClass
text classification
zero-shot
small language models
RAG
sentiment analysis
Instructions to use knowledgator/gliclass-base-v1.0-init with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use knowledgator/gliclass-base-v1.0-init with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("zero-shot-classification", model="knowledgator/gliclass-base-v1.0-init")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("knowledgator/gliclass-base-v1.0-init", dtype="auto") - Notebooks
- Google Colab
- Kaggle
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-base-v1")
|
| 38 |
-
tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-base-v1")
|
| 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-base-v1.0-init")
|
| 40 |
+
tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-base-v1.0-init")
|
| 41 |
|
| 42 |
pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')
|
| 43 |
|