Fixed bug tokenizer SecureBERT

#1
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -65,7 +65,7 @@ The model was evaluated on the held-out test set using standard multi-label clas
65
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
66
  import torch
67
 
68
- tokenizer = AutoTokenizer.from_pretrained("Sana9/securebert-vuln2attack-flat")
69
  model = AutoModelForSequenceClassification.from_pretrained("Sana9/securebert-vuln2attack-flat")
70
 
71
  text = "Buffer overflow vulnerability in OpenSSL allows remote attackers to execute arbitrary code."
 
65
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
66
  import torch
67
 
68
+ tokenizer = AutoTokenizer.from_pretrained("Sana9/securebert-vuln2attack-flat", use_fast=False)
69
  model = AutoModelForSequenceClassification.from_pretrained("Sana9/securebert-vuln2attack-flat")
70
 
71
  text = "Buffer overflow vulnerability in OpenSSL allows remote attackers to execute arbitrary code."