Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

ayshajavd
/
graphcodebert-vuln-classifier

Text Classification
Transformers
Safetensors
English
code
roberta
security
vulnerability-detection
code-analysis
multi-label-classification
graphcodebert
owasp
cwe
static-analysis
Eval Results (legacy)
text-embeddings-inference
Model card Files Files and versions
xet
Community

Instructions to use ayshajavd/graphcodebert-vuln-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use ayshajavd/graphcodebert-vuln-classifier with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("text-classification", model="ayshajavd/graphcodebert-vuln-classifier")
    # Load model directly
    from transformers import AutoTokenizer, AutoModelForSequenceClassification
    
    tokenizer = AutoTokenizer.from_pretrained("ayshajavd/graphcodebert-vuln-classifier")
    model = AutoModelForSequenceClassification.from_pretrained("ayshajavd/graphcodebert-vuln-classifier")
  • Notebooks
  • Google Colab
  • Kaggle
graphcodebert-vuln-classifier
338 MB
Ctrl+K
Ctrl+K
  • 1 contributor
History: 4 commits
ayshajavd's picture
ayshajavd
Fix id2label mapping with actual CWE names
64e4ac8 verified about 1 month ago
  • .gitattributes
    1.52 kB
    initial commit about 1 month ago
  • README.md
    4.17 kB
    GraphCodeBERT vulnerability classifier (CPU-trained, 20K balanced subset) about 1 month ago
  • config.json
    1.98 kB
    Fix id2label mapping with actual CWE names about 1 month ago
  • label_config.json
    5.76 kB
    Upload label_config.json with huggingface_hub about 1 month ago
  • model.safetensors
    334 MB
    xet
    GraphCodeBERT vulnerability classifier (CPU-trained, 20K balanced subset) about 1 month ago
  • tokenizer.json
    3.71 MB
    GraphCodeBERT vulnerability classifier (CPU-trained, 20K balanced subset) about 1 month ago
  • tokenizer_config.json
    406 Bytes
    GraphCodeBERT vulnerability classifier (CPU-trained, 20K balanced subset) about 1 month ago
  • training_args.bin

    Detected Pickle imports (10)

    • "transformers.trainer_utils.SaveStrategy",
    • "transformers.training_args.OptimizerNames",
    • "transformers.trainer_utils.SchedulerType",
    • "transformers.trainer_utils.HubStrategy",
    • "transformers.trainer_pt_utils.AcceleratorConfig",
    • "transformers.training_args.TrainingArguments",
    • "torch.device",
    • "accelerate.utils.dataclasses.DistributedType",
    • "transformers.trainer_utils.IntervalStrategy",
    • "accelerate.state.PartialState"

    How to fix it?

    5.33 kB
    xet
    GraphCodeBERT vulnerability classifier (CPU-trained, 20K balanced subset) about 1 month ago