metadata
license: mit
language:
- en
base_model:
- google-bert/bert-large-cased
pipeline_tag: token-classification
library_name: transformers
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)