Downloads · 30 days
6
2% of all-time downloads
hadiaskari98/Vulnerability_NER_prod
Vulnerability_NER_prod is a token classification model from hadiaskari98. Use it when you need labels on individual words, such as names. It is set up for transformers. The card lists the license as mit.
Downloads · 30 days
6
2% of all-time downloads
All-time downloads
290
Public
Parameters
333M
2.7 GB on disk
Likes
0
Public
Click a slice to open those files.
.bin1.3 GB · 50%
From the Hugging Face model README
How to use
tokenizer = AutoTokenizer.from_pretrained("hadiaskari98/Vulnerability_NER_prod")
model = AutoModelForTokenClassification.from_pretrained("hadiaskari98/Vulnerability_NER_prod")
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
example = "This is an example of a SQL Injection attack"
ner_results = nlp(example)
print(ner_results)