govtech/lionguard-v1

By govtech · ↓ 14 · ♥ 10 · Updated 2024-11-16

What's in this repo

This page lists the files in govtech/lionguard-v1, including lionguard-binary.onnx, lionguard-harassment.onnx, lionguard-hateful.onnx and more.

8 ONNX models (.onnx files)
14 files, including 8 .onnx, 1 .md, 1 .json, 1 .py
Last updated 2024-11-16

ONNX files

File	Size	Download
models/lionguard-binary.onnx	—	Download
models/lionguard-harassment.onnx	—	Download
models/lionguard-hateful.onnx	—	Download
models/lionguard-public_harm.onnx	—	Download
models/lionguard-self_harm.onnx	—	Download
models/lionguard-sexual.onnx	—	Download
models/lionguard-toxic.onnx	—	Download
models/lionguard-violent.onnx	—	Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

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

Install transformers , onnxruntime and huggingface_hub libraries.

pip install transformers onnxruntime huggingface_hub

Run inference.

python inference.py '["Example text 1"]'

Technical report is available here.

Model info

transformers

transformers onnx arxiv:2407.10995 license:other endpoints_compatible region:us

↓ 14 ♥ 10

Created: 2024-07-16

Updated: 2024-11-16

View on Hugging Face

Files (14)

.gitattributes

.gitignore

LICENSE

README.md

config.json

inference.py

models/lionguard-binary.onnx ONNX

models/lionguard-harassment.onnx ONNX

models/lionguard-hateful.onnx ONNX

models/lionguard-public_harm.onnx ONNX

models/lionguard-self_harm.onnx ONNX

models/lionguard-sexual.onnx ONNX

models/lionguard-toxic.onnx ONNX

models/lionguard-violent.onnx ONNX

Related models

More models

No task type declared.