CoolFace
Apppublic

martynattakit/CodeSentinel-CWE_Classification

sourceHugging Faceupdated 4mo agoView on Hugging Face
1likes
__init__.py19 linesDownload Raw Back to pipeline
1"""2pipeline/3Vulnerability classification pipeline.4 5Components:6    classifier.py    — RoBERTa CWE classifier (text → CWE ID)7    code_analyzer.py — Qwen code analyzer (code → NL description)8    atlas_matcher.py — ATLAS pattern matcher (text → ATLAS technique)9    router.py        — Main router (raw input → unified output card)10 11Usage:12    from pipeline.router import route13    result = route("def get_user(name): return db.execute('SELECT * FROM users WHERE name=' + name)")14"""15 16from pipeline.router import route, get_router17 18__all__ = ["route", "get_router"]19