Nitishkumar-ai commited on
Commit
4fb819f
·
verified ·
1 Parent(s): e4f3d12

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. README_SUBMISSION.md +2 -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
  ![Accuracy Comparison](plots/baseline_vs_trained.png)
29
- *Our trained agent improved detection accuracy from **X%** (baseline) to **Y%**.*
30
 
31
  ### 3. Per-CWE Breakdown
32
  ![CWE Breakdown](plots/per_cwe.png)
@@ -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):** [Link](<LINK_TO_HF_SPACE>)
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
  ![Accuracy Comparison](plots/baseline_vs_trained.png)
29
+ *Our trained agent improved detection accuracy from **50%** (baseline) to **74%**.*
30
 
31
  ### 3. Per-CWE Breakdown
32
  ![CWE Breakdown](plots/per_cwe.png)
 
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 OpenEnv RL environment for commit-time vuln detection (hackathon submission)"
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"