Add pipeline tag and library name
Browse filesThis 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.
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
|