Instructions to use mahdin70/codebert-devign-code-vulnerability-detector with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mahdin70/codebert-devign-code-vulnerability-detector with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="mahdin70/codebert-devign-code-vulnerability-detector")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("mahdin70/codebert-devign-code-vulnerability-detector") model = AutoModelForSequenceClassification.from_pretrained("mahdin70/codebert-devign-code-vulnerability-detector") - Notebooks
- Google Colab
- Kaggle
Are those models linked to any published papers at all?
Hello,
I'm looking into current models that have been fined tuned for vulnerability detection for different languages and I was wondering if you can provide any research you've done or any papers linked to these trained models so I can reference them?
Hi, All Models and Datasets under my profile now have a DOI associated with them. You can cite them accordingly.
Click the "Three Dots" icon beside the "Deploy" button at the top of the model card. You will find the "Cite this Model" option in that menu.
You will find something like this :
@misc {mukit_mahdin_2025,
author = { Mukit Mahdin },
title = { codebert-devign-code-vulnerability-detector (Revision 7c73292) },
year = 2025,
url = { https://huggingface.co/mahdin70/codebert-devign-code-vulnerability-detector },
doi = { 10.57967/hf/7177 },
publisher = { Hugging Face }
}