cisco-ehsan commited on
Commit
f260674
·
verified ·
1 Parent(s): 5bed2dc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -28,11 +28,11 @@ It is fine-tuned for **code-level security analysis**, extending the capabilitie
28
  - **Number of labels:** 2
29
  - **Language:** English (source code tokens)
30
  - **License:** Apache-2.0
31
- - **Finetuned from model:** [CiscoAITeam/SecureBERT2.0-base](https://huggingface.co/CiscoAITeam/SecureBERT2.0-base)
32
 
33
  ### Model Sources
34
 
35
- - **Repository:** [https://huggingface.co/CiscoAITeam/SecureBERT2.0-code-vuln-detection](https://huggingface.co/CiscoAITeam/SecureBERT2.0-code-vuln-detection)
36
  - **Paper:** [arXiv:2510.00240](https://arxiv.org/abs/2510.00240)
37
 
38
  ---
@@ -80,7 +80,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
80
  import torch
81
 
82
  # Path to the model
83
- model_dir = "CiscoAITeam/SecureBERT2.0-code-vuln-detection"
84
 
85
  # Load tokenizer and model
86
  tokenizer = AutoTokenizer.from_pretrained(model_dir)
 
28
  - **Number of labels:** 2
29
  - **Language:** English (source code tokens)
30
  - **License:** Apache-2.0
31
+ - **Finetuned from model:** [cisco-ai/SecureBERT2.0-base](https://huggingface.co/cisco-ai/SecureBERT2.0-base)
32
 
33
  ### Model Sources
34
 
35
+ - **Repository:** [https://huggingface.co/cisco-ai/SecureBERT2.0-code-vuln-detection](https://huggingface.co/cisco-ai/SecureBERT2.0-code-vuln-detection)
36
  - **Paper:** [arXiv:2510.00240](https://arxiv.org/abs/2510.00240)
37
 
38
  ---
 
80
  import torch
81
 
82
  # Path to the model
83
+ model_dir = "cisco-ai/SecureBERT2.0-code-vuln-detection"
84
 
85
  # Load tokenizer and model
86
  tokenizer = AutoTokenizer.from_pretrained(model_dir)