Zeqh's picture
Fix README metadata emoji
55e1a5a
---
title: Mental Health Text Classifier (BERT Final v1)
emoji: "🧠"
colorFrom: blue
colorTo: pink
sdk: gradio
sdk_version: 4.44.0
app_file: app.py
pinned: false
license: mit
tags:
- text-classification
- mental-health
- bert
- pytorch
- transformers
---
# Mental Health Text Classifier (BERT Final v1)
This Space serves a fine-tuned BERT model that classifies text into three categories:
- Normal
- Distressed
- Suicidal
## Model
- Architecture: BERT-base fine-tuned for sequence classification
- Framework: PyTorch + Transformers
- Labels: 0 = Normal, 1 = Distressed, 2 = Suicidal
## Usage
Enter text in the input box and click Analyze Text to see the predicted class and class probabilities.
## Disclaimer
This tool is for research and educational purposes only and should not be used as a substitute for
professional mental health care. If you or someone you know is experiencing a mental health crisis,
please contact a mental health professional or crisis helpline immediately.