Spaces:
Configuration error
Configuration error
Upload folder using huggingface_hub
Browse files- README_SUBMISSION.md +2 -2
- pyproject.toml +35 -36
README_SUBMISSION.md
CHANGED
|
@@ -26,7 +26,7 @@ We trained **Llama-3.2-3B-Instruct** using **GRPO** via TRL and Unsloth.
|
|
| 26 |
|
| 27 |
### 2. Detection Accuracy: Baseline vs. Trained
|
| 28 |

|
| 29 |
-
*Our trained agent improved detection accuracy from **
|
| 30 |
|
| 31 |
### 3. Per-CWE Breakdown
|
| 32 |

|
|
@@ -37,7 +37,7 @@ We trained **Llama-3.2-3B-Instruct** using **GRPO** via TRL and Unsloth.
|
|
| 37 |
*Watch as a trained CommitGuard agent requests context to identify a complex privilege escalation vulnerability that the baseline model missed.*
|
| 38 |
|
| 39 |
## Links
|
| 40 |
-
- **HF Space (Env):** [
|
| 41 |
- **Training Notebook:** [Link](<LINK_TO_NOTEBOOK>)
|
| 42 |
- **W&B Training Logs:** [Link](<LINK_TO_WANDB>)
|
| 43 |
- **HF Blog Post:** [Link](<LINK_TO_BLOG>)
|
|
|
|
| 26 |
|
| 27 |
### 2. Detection Accuracy: Baseline vs. Trained
|
| 28 |

|
| 29 |
+
*Our trained agent improved detection accuracy from **50%** (baseline) to **74%**.*
|
| 30 |
|
| 31 |
### 3. Per-CWE Breakdown
|
| 32 |

|
|
|
|
| 37 |
*Watch as a trained CommitGuard agent requests context to identify a complex privilege escalation vulnerability that the baseline model missed.*
|
| 38 |
|
| 39 |
## Links
|
| 40 |
+
- **HF Space (Env):** [https://huggingface.co/spaces/Nitishkumar-ai/commitguard](https://huggingface.co/spaces/Nitishkumar-ai/commitguard)
|
| 41 |
- **Training Notebook:** [Link](<LINK_TO_NOTEBOOK>)
|
| 42 |
- **W&B Training Logs:** [Link](<LINK_TO_WANDB>)
|
| 43 |
- **HF Blog Post:** [Link](<LINK_TO_BLOG>)
|
pyproject.toml
CHANGED
|
@@ -1,36 +1,35 @@
|
|
| 1 |
-
[project]
|
| 2 |
-
name = "commitguard"
|
| 3 |
-
version = "0.1.0"
|
| 4 |
-
description = "CommitGuard
|
| 5 |
-
readme = "README.md"
|
| 6 |
-
requires-python = ">=3.10"
|
| 7 |
-
dependencies = [
|
| 8 |
-
"fastapi>=0.110",
|
| 9 |
-
"uvicorn[standard]>=0.27",
|
| 10 |
-
"pydantic>=2.6",
|
| 11 |
-
"openenv>=0.1.13",
|
| 12 |
-
]
|
| 13 |
-
|
| 14 |
-
[project.optional-dependencies]
|
| 15 |
-
train = [
|
| 16 |
-
"requests",
|
| 17 |
-
"torch",
|
| 18 |
-
"transformers",
|
| 19 |
-
"trl",
|
| 20 |
-
"accelerate",
|
| 21 |
-
"peft",
|
| 22 |
-
"datasets",
|
| 23 |
-
"wandb",
|
| 24 |
-
"matplotlib",
|
| 25 |
-
]
|
| 26 |
-
|
| 27 |
-
[project.scripts]
|
| 28 |
-
server = "commitguard_env.server:main"
|
| 29 |
-
|
| 30 |
-
[tool.setuptools]
|
| 31 |
-
packages = ["commitguard_env"]
|
| 32 |
-
|
| 33 |
-
[build-system]
|
| 34 |
-
requires = ["setuptools>=68"]
|
| 35 |
-
build-backend = "setuptools.build_meta"
|
| 36 |
-
|
|
|
|
| 1 |
+
[project]
|
| 2 |
+
name = "commitguard"
|
| 3 |
+
version = "0.1.0"
|
| 4 |
+
description = "CommitGuard OpenEnv RL environment for commit-time vuln detection"
|
| 5 |
+
readme = "README.md"
|
| 6 |
+
requires-python = ">=3.10"
|
| 7 |
+
dependencies = [
|
| 8 |
+
"fastapi>=0.110",
|
| 9 |
+
"uvicorn[standard]>=0.27",
|
| 10 |
+
"pydantic>=2.6",
|
| 11 |
+
"openenv>=0.1.13",
|
| 12 |
+
]
|
| 13 |
+
|
| 14 |
+
[project.optional-dependencies]
|
| 15 |
+
train = [
|
| 16 |
+
"requests",
|
| 17 |
+
"torch",
|
| 18 |
+
"transformers",
|
| 19 |
+
"trl",
|
| 20 |
+
"accelerate",
|
| 21 |
+
"peft",
|
| 22 |
+
"datasets",
|
| 23 |
+
"wandb",
|
| 24 |
+
"matplotlib",
|
| 25 |
+
]
|
| 26 |
+
|
| 27 |
+
[project.scripts]
|
| 28 |
+
server = "commitguard_env.server:main"
|
| 29 |
+
|
| 30 |
+
[tool.setuptools]
|
| 31 |
+
packages = ["commitguard_env"]
|
| 32 |
+
|
| 33 |
+
[build-system]
|
| 34 |
+
requires = ["setuptools>=68"]
|
| 35 |
+
build-backend = "setuptools.build_meta"
|
|
|