File size: 541 Bytes
64ac508
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
---
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")