Upload folder using huggingface_hub

README.md

@@ -1,5 +1,5 @@
 ---
-title: Mental Health Triage Assistant
+title: Undertone
 emoji: 🧠
 colorFrom: blue
 colorTo: red
@@ -8,4 +8,12 @@ sdk_version: 5.33.0
 app_file: app.py
 pinned: false
 license: mit
+short_description: mental health message triage.
 ---
+
+# 🧠 Undertone
+
+**The signal beneath the surface.**
+
+A fine-tuned DistilBERT classifier that triages mental health support messages into three categories with a focus on surfacing subtle distress cues.
+

app.py

@@ -1,6 +1,7 @@
 """
-app.py — Mental Health Triage Assistant
-Gradio interface，部署到 HuggingFace Spaces
+app.py — Undertone
+The signal beneath the surface.
+Mental Health Triage Assistant · Gradio interface for HuggingFace Spaces
 """
 
 import torch
@@ -11,7 +12,6 @@ from transformers import (
     DistilBertForSequenceClassification,
 )
 
-# ── 配置 ─────────────────────────────────────────────────
 MODEL_DIR = Path(__file__).parent / "model" / "saved"
 
 LABEL2ID = {"informational": 0, "emotional_support": 1, "escalation": 2}
@@ -210,13 +210,13 @@ body, .gradio-container {
 """
 
 # ── 界面 ─────────────────────────────────────────────────
-with gr.Blocks(css=CUSTOM_CSS, title="Mental Health Triage") as demo:
+with gr.Blocks(css=CUSTOM_CSS, title="Undertone") as demo:
 
     gr.HTML("""
     <div class="app-header">
-        <h1>🧠 Mental Health Triage Assistant</h1>
-        <p>Classifies support messages to help route them to the right response level</p>
-        <span class="badge">Research Demo · DistilBERT · Duke AIPI 561</span>
+        <h1>🧠 Undertone</h1>
+        <p>The signal beneath the surface — mental health message triage</p>
+        <span class="badge">Research Demo · DistilBERT Fine-Tuned</span>
     </div>
     """)