nielsr HF staff commited on
Commit
00c0e13
·
verified ·
1 Parent(s): 2972348

Add pipeline tag and library name

Browse files

This PR adds the `pipeline_tag` and `library_name` to the model card metadata to improve discoverability and clarity. The `pipeline_tag` is set to `text-classification` reflecting the model's functionality. The `library_name` is set to `transformers` as indicated by the provided usage example.

Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -2,6 +2,8 @@
2
  license: apache-2.0
3
  base_model:
4
  - Qwen/Qwen2.5-1.5B
 
 
5
  ---
6
 
7
  ## Overview
@@ -98,4 +100,6 @@ print(f"Overall Prompt Classification => {'UNSAFE' if overall_label == 1 else 'S
98
  primaryClass={cs.CL},
99
  url={https://arxiv.org/abs/2502.05163},
100
  }
101
- ```
 
 
 
2
  license: apache-2.0
3
  base_model:
4
  - Qwen/Qwen2.5-1.5B
5
+ pipeline_tag: text-classification
6
+ library_name: transformers
7
  ---
8
 
9
  ## Overview
 
100
  primaryClass={cs.CL},
101
  url={https://arxiv.org/abs/2502.05163},
102
  }
103
+ ```
104
+
105
+ Code: https://github.com/yihedeng9/DuoGuard