Update README.md
Browse files
README.md
CHANGED
@@ -52,7 +52,7 @@ from gliclass import GLiClassModel, ZeroShotClassificationPipeline
|
|
52 |
from transformers import AutoTokenizer
|
53 |
|
54 |
model = GLiClassModel.from_pretrained("knowledgator/gliclass-modern-large-v2.0-init")
|
55 |
-
tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-modern-large-v2.0-init")
|
56 |
pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')
|
57 |
|
58 |
text = "One day I will see the world!"
|
|
|
52 |
from transformers import AutoTokenizer
|
53 |
|
54 |
model = GLiClassModel.from_pretrained("knowledgator/gliclass-modern-large-v2.0-init")
|
55 |
+
tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-modern-large-v2.0-init", add_prefix_space=True)
|
56 |
pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')
|
57 |
|
58 |
text = "One day I will see the world!"
|