LionGuard π¦
Collection
A Singapore-contextualized moderation classifier. β’ 2 items β’ Updated β’ 1
How to use govtech/lionguard-v1 with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("govtech/lionguard-v1", dtype="auto")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.
transformers , onnxruntime and huggingface_hub libraries.pip install transformers onnxruntime huggingface_hub
python inference.py '["Example text 1"]'
Technical report is available here.
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("govtech/lionguard-v1", dtype="auto")