π¨ Urgency Model Aura (KS-Vijay)
This is a DistilBERT-based text classification model trained to detect the urgency level of textual grievances submitted by startups. It's part of an AI-based Grievance Redressal Platform for startups.
π§ Use Case
Helps categorize complaints into urgency levels:
Low
Medium
High
Critical
This allows startups or organizations to prioritize tickets and respond efficiently based on severity.
π How It Works
The model takes in a complaint text and outputs a classification label. It was trained on labeled grievance data using the π€ transformers
and datasets
libraries.
π¦ Model Details
- Architecture:
DistilBERT
- Framework:
PyTorch
- Model type:
Text Classification
- Format:
safetensors
- Dataset: Custom
complaints.csv
(internal) - Labels: Urgency levels (Critical, High, Medium, Low)
π Example Input
"My startup's payment system has been offline since morning!"
- Downloads last month
- 14
Model tree for KS-Vijay/urgency-model-aura
Base model
distilbert/distilbert-base-uncased