|
--- |
|
license: other |
|
license_name: govtech-singapore |
|
license_link: LICENSE |
|
--- |
|
|
|
# LionGuard |
|
|
|
LionGuard is a classifier for detecting unsafe content in the Singapore context. It uses pre-trained BAAI English embeddings and performs classification with a trained Ridge Classification model. |
|
|
|
# Usage |
|
|
|
1. Install `transformers` , `onnxruntime` and `huggingface_hub` libraries. |
|
``` |
|
pip install transformers onnxruntime huggingface_hub |
|
``` |
|
|
|
2. Run inference. |
|
``` |
|
python inference.py '["Example text 1"]' |
|
``` |
|
|
|
Technical report is available [here](https://arxiv.org/abs/2407.10995). |
|
|
