knowledgator
/

gliclass-small-v1.0-init

Zero-Shot Classification

text classification

small language models

sentiment analysis

Model card Files Files and versions Community

BioMike commited on Jun 3, 2024

Commit

06b9c34

·

verified ·

1 Parent(s): 3a5afb0

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -34,8 +34,8 @@ Than you need to initialize a model and a pipeline:
 from gliclass import GLiClassModel, ZeroShotClassificationPipeline
 from transformers import AutoTokenizer
-model = GLiClassModel.from_pretrained("knowledgator/gliclass-small-v1")
-tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-small-v1")
 pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')

 from gliclass import GLiClassModel, ZeroShotClassificationPipeline
 from transformers import AutoTokenizer
+model = GLiClassModel.from_pretrained("knowledgator/gliclass-small-v1.0")
+tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-small-v1.0")
 pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')