| --- |
| license: apache-2.0 |
| language: |
| - en |
| tags: |
| - aisafety |
| - onnx |
| - confidence |
| - risk-classifier |
| - offline-ai |
| --- |
| # π§ TrustFrame ONNX Modules |
|
|
| Pre-trained lightweight models for AI safety: |
|
|
| - `confidence_tinybert.onnx` β Estimates confidence score (0.0β1.0) |
| - `risk_classifier_distilbert.onnx` β Classifies risk domains (medical, legal, code, financial) |
| - `risk_label_map.json` β Maps class IDs to domain names |
|
|
| ## π₯ Usage |
|
|
| ```python |
| from onnxruntime import InferenceSession |
| |
| session = InferenceSession("confidence_tinybert.onnx") |