Views
No views yet
1
2tokenizer = AutoTokenizer.from_pretrained("hadiaskari98/Vulnerability_NER_prod")
3model = AutoModelForTokenClassification.from_pretrained("hadiaskari98/Vulnerability_NER_prod")
4
5nlp = pipeline("ner", model=model, tokenizer=tokenizer)
6
7example = "This is an example of a SQL Injection attack"
8
9ner_results = nlp(example)
10print(ner_results)