|
|
--- |
|
|
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") |
