| | --- |
| | license: mit |
| | language: |
| | - en |
| | base_model: |
| | - google-bert/bert-large-cased |
| | pipeline_tag: token-classification |
| | library_name: transformers |
| | --- |
| | |
| |
|
| | **How to use** |
| |
|
| |
|
| | ```from transformers import AutoTokenizer, AutoModelForTokenClassification |
| | |
| | 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) |
| | |
