urgency-model-aura / README.md
KS-Vijay's picture
Update README.md
e56f215 verified
metadata
language: en
license: apache-2.0
base_model: distilbert-base-uncased
tags:
  - text-classification
  - urgency-detection
  - startup-grievances
  - transformers
  - huggingface
pipeline_tag: text-classification
widget:
  - text: Our server is down! We need immediate help!

🚨 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!"