Downloads · 30 days
2
8% of all-time downloads
tk-prompt-prune
VisheshKJha/tk-prompt-prune
tk-prompt-prune is a machine learning model from VisheshKJha. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Unified DistilBERT model performing 4 tasks: 1. Spell Checker 2. Framework Matcher 3. Sensitive Detector (with 83+ pattern-based detection) 4. Intent Classifier
All-time downloads
26
Public
Repo size
1.9 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.data1.6 GB · 83%
At a glance
- Model type
- distilbert
- Access
- Public
- Created
- Dec 1, 2025
- Updated
- Dec 15, 2025
- SHA
- 388e2c77
Model card
From the Hugging Face model 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
- Type
- distilbert
- Created
- Dec 1, 2025
- Updated
- Dec 15, 2025