Ihor commited on
Commit
6ccb923
·
verified ·
1 Parent(s): 96340b1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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!"