Update README.md
Browse files
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:** [
|
| 32 |
|
| 33 |
### Model Sources
|
| 34 |
|
| 35 |
-
- **Repository:** [https://huggingface.co/
|
| 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 = "
|
| 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)
|