VisheshKJha/tk-prompt-prune
By VisheshKJha · ↓ 0 · ♥ 0 · Updated 2025-12-15
What's in this repo
This page lists the files in VisheshKJha/tk-prompt-prune, including model.onnx, model_quantized.onnx and more.
- 8 ONNX models (.onnx files)
- 29 files, including 8 .json, 8 .onnx, 6 .data, 2 .md, 2 .js, 1 .bin
- Last updated 2025-12-15
ONNX files
| File | Size | Download |
|---|---|---|
| encoder/model.onnx | — | Download |
| encoder/model_quantized.onnx | — | Download |
| framework_matcher/model.onnx | — | Download |
| intent_classifier/model.onnx | — | Download |
| model.onnx | — | Download |
| model_quantized.onnx | — | Download |
| sensitive_detector/model.onnx | — | Download |
| spell_checker/model.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
PromptPrune Unified Multi-Task Model
Overview
Unified DistilBERT model performing 4 tasks:
- Spell Checker
- Framework Matcher
- Sensitive Detector (with 83+ pattern-based detection)
- Intent Classifier
Comprehensive PII Detection
This model includes 83+ sensitive data patterns that run BEFORE model inference:
- 50+ PII types (SSN, PAN, Aadhaar, Credit Cards, etc.)
- 20+ Non-PII sensitive data (Passwords, Private Keys, Biometric, Genetic, etc.)
- Special Category Data (GDPR Article 9)
Usage
See browser_inference_example.js for browser extension integration.
Files
model.onnx- ONNX model for inferencemodel_quantized.onnx- Quantized ONNX modelpii_patterns.js- Comprehensive PII detection patternsbrowser_inference_example.js- Browser inference example
Model info
Created: 2025-12-01
Updated: 2025-12-15
View on Hugging FaceFiles (29)
Related models
More from VisheshKJha
No other models by this author.
More models
No task type declared.