Downloads · 30 days
1
1% of all-time downloads
sage-mental-health-chatbot-model
yusrawaheedyw79/sage-mental-health-chatbot-model
sage-mental-health-chatbot-model is a machine learning model from yusrawaheedyw79. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This directory contains the fine-tuned DistilGPT2 model for empathetic mental health support.
All-time downloads
112
Public
Parameters
81.9M
328 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors328 MB · 99%
At a glance
- Model type
- gpt2
- Access
- Public
- Created
- Jun 27, 2026
- Updated
- Jun 27, 2026
- SHA
- 4089f823
Model card
From the Hugging Face model README
Fine-Tuned DistilGPT2 Model
This directory contains the fine-tuned DistilGPT2 model for empathetic mental health support.
Getting the model weights
The model weights (model.safetensors, ~320 MB) are not stored in git due to size.
Option 1: Download from Hugging Face (coming soon)
The model may be uploaded to Hugging Face Model Hub.
Option 2: Train it yourself in Colab
- Open
notebook/finetune_mental_health.ipynbin Google Colab - Set runtime to T4 GPU (Runtime → Change runtime type)
- Run all cells (~20 minutes)
- The model will save to this directory automatically
Option 3: Use the base model (fallback)
The app (app.py) automatically falls back to the base distilgpt2 if the fine-tuned weights are missing — you can run the chat interface right away without fine-tuning.
Files
config.json— Model configurationgeneration_config.json— Generation parameterstokenizer.json— Tokenizer vocabularytokenizer_config.json— Tokenizer configurationREADME.md— This file
- Type
- gpt2
- Created
- Jun 27, 2026
- Updated
- Jun 27, 2026