Merge branch 'main' of hf.co:knowledgator/gliclass-modern-base-v2.0-init into main
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-base-v2.0-init")
|
| 55 |
-
tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-modern-base-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-base-v2.0-init")
|
| 55 |
+
tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-modern-base-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!"
|