Update README.md
Browse files
README.md
CHANGED
|
@@ -4,7 +4,6 @@ This is a ModernBERT model fine-tuned for **Code Vulnerability Detection**.
|
|
| 4 |
It is built on top of **SecureBERT 2.0**.
|
| 5 |
|
| 6 |
## Model Details
|
| 7 |
-
- Original checkpoint: `/teamspace/studios/this_studio/secure_modern_bert/sentiment_classif/checkpoint_epoch_10.pth`
|
| 8 |
- Model type: `classification`
|
| 9 |
- Number of labels: 2
|
| 10 |
- Architecture: ModernBertForSequenceClassification
|
|
@@ -13,8 +12,8 @@ It is built on top of **SecureBERT 2.0**.
|
|
| 13 |
## Usage Example
|
| 14 |
```python
|
| 15 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 16 |
-
tokenizer = AutoTokenizer.from_pretrained('/
|
| 17 |
-
model = AutoModelForSequenceClassification.from_pretrained('/
|
| 18 |
```
|
| 19 |
|
| 20 |
## Notes
|
|
|
|
| 4 |
It is built on top of **SecureBERT 2.0**.
|
| 5 |
|
| 6 |
## Model Details
|
|
|
|
| 7 |
- Model type: `classification`
|
| 8 |
- Number of labels: 2
|
| 9 |
- Architecture: ModernBertForSequenceClassification
|
|
|
|
| 12 |
## Usage Example
|
| 13 |
```python
|
| 14 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 15 |
+
tokenizer = AutoTokenizer.from_pretrained('CiscoAITeam/SecureBERT2.0-code-vuln-detection')
|
| 16 |
+
model = AutoModelForSequenceClassification.from_pretrained('CiscoAITeam/SecureBERT2.0-code-vuln-detection')
|
| 17 |
```
|
| 18 |
|
| 19 |
## Notes
|