Instructions to use cisco-ai/SecureBERT2.0-code-vuln-detection with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use cisco-ai/SecureBERT2.0-code-vuln-detection with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="cisco-ai/SecureBERT2.0-code-vuln-detection")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("cisco-ai/SecureBERT2.0-code-vuln-detection") model = AutoModelForSequenceClassification.from_pretrained("cisco-ai/SecureBERT2.0-code-vuln-detection") - Inference
- Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -22,7 +22,7 @@ It leverages cybersecurity-aware representations learned by SecureBERT 2.0 and a
|
|
| 22 |
This model classifies source code snippets as either **vulnerable** or **non-vulnerable** using the ModernBERT architecture.
|
| 23 |
It is fine-tuned for **code-level security analysis**, extending the capabilities of SecureBERT 2.0.
|
| 24 |
|
| 25 |
-
- **Developed by:** Cisco AI
|
| 26 |
- **Model type:** Sequence classification
|
| 27 |
- **Architecture:** `ModernBertForSequenceClassification`
|
| 28 |
- **Number of labels:** 2
|
|
@@ -202,9 +202,9 @@ Carbon footprint can be estimated using the [Machine Learning Impact Calculator]
|
|
| 202 |
|
| 203 |
## Model Card Authors
|
| 204 |
|
| 205 |
-
Cisco AI
|
| 206 |
|
| 207 |
## Model Card Contact
|
| 208 |
|
| 209 |
-
For inquiries, please contact [Cisco AI
|
| 210 |
|
|
|
|
| 22 |
This model classifies source code snippets as either **vulnerable** or **non-vulnerable** using the ModernBERT architecture.
|
| 23 |
It is fine-tuned for **code-level security analysis**, extending the capabilities of SecureBERT 2.0.
|
| 24 |
|
| 25 |
+
- **Developed by:** Cisco AI
|
| 26 |
- **Model type:** Sequence classification
|
| 27 |
- **Architecture:** `ModernBertForSequenceClassification`
|
| 28 |
- **Number of labels:** 2
|
|
|
|
| 202 |
|
| 203 |
## Model Card Authors
|
| 204 |
|
| 205 |
+
Cisco AI
|
| 206 |
|
| 207 |
## Model Card Contact
|
| 208 |
|
| 209 |
+
For inquiries, please contact [Cisco AI](mailto:eaghaei@cisco.com)
|
| 210 |
|