Zeqh's picture
Fix README metadata emoji
55e1a5a

A newer version of the Gradio SDK is available: 6.15.2

Upgrade
metadata
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.