Text Classification
Transformers
Safetensors
code
roberta
codebert
vulnerability-detection
cybersecurity
software-security
static-analysis
text-embeddings-inference
Instructions to use Khansa-saAI-29/final-codebert-model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Khansa-saAI-29/final-codebert-model with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Khansa-saAI-29/final-codebert-model")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Khansa-saAI-29/final-codebert-model") model = AutoModelForSequenceClassification.from_pretrained("Khansa-saAI-29/final-codebert-model", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Fine-tuned CodeBERT for Vulnerability Detection
|
| 2 |
+
|
| 3 |
+
## Description
|
| 4 |
+
|
| 5 |
+
This model is fine-tuned on the Big-Vul dataset for binary vulnerability classification.
|
| 6 |
+
|
| 7 |
+
Labels:
|
| 8 |
+
- 0 = Non-Vulnerable
|
| 9 |
+
- 1 = Vulnerable
|
| 10 |
+
|
| 11 |
+
Base model:
|
| 12 |
+
microsoft/codebert-base
|
| 13 |
+
|
| 14 |
+
Framework:
|
| 15 |
+
Transformers + PyTorch
|