Upload folder using huggingface_hub

README.md

@@ -9,6 +9,8 @@ tags:
   - distilbert
   - patient-doctor-conversations
   - text-classification
+  - clinical-AI
+  - mental-health
 model_index:
   - name: patient-emotion-classifier
     results:
@@ -16,34 +18,61 @@ model_index:
           type: text-classification
         metrics:
           - type: accuracy
-            value: 0.67
+            value: 0.713
           - type: f1
-            value: 0.62
+            value: 0.722
 ---
 
-# Patient Emotion Analysis
+<div align="center">
 
-NLP emotion classification model for patient-doctor conversations.
+# 🤖 Patient Emotion Classifier
 
-## Emotion Categories
+**Advanced AI-Powered Emotion Recognition for Healthcare Dialogues**
+
+*Part of the Blended AI+X Initiative — Bridging Artificial Intelligence and Healthcare*
+
+---
+
+[![Model](https://img.shields.io/badge/Model-DistilBERT-blue)](https://huggingface.co/distilbert/distilbert-base-uncased)
+[![License](https://img.shields.io/badge/License-Apache--2.0-green)](LICENSE)
+[![Performance](https://img.shields.io/badge/F1--Score-72.2%25-orange)]()
+
+</div>
+
+## 🔬 Overview
+
+We are thrilled to introduce **Patient Emotion Classifier**, a state-of-the-art NLP model engineered to understand emotional nuances in patient-doctor conversations. 
+
+This model represents our commitment to advancing **AI for Healthcare (AI+X)**, leveraging cutting-edge transformer architectures to bridge the gap between artificial intelligence and compassionate care.
+
+### Key Capabilities
+
+- **Multiclass Emotion Recognition** — Identifies 6 distinct emotional states in clinical dialogues
+- **Healthcare-Optimized** — Specifically trained on medical conversation data
+- **Production-Ready** — Deployable via REST API for real-time inference
+- **Lightweight & Efficient** — Built on DistilBERT for fast inference
+
+## 🎯 Emotion Categories
+
+Our model classifies emotional states into **6 clinically-relevant categories**:
 
 | Category | Description |
 |----------|-------------|
-| Neutral | Neutral statements |
-| Anxiety/Fear | Patient expresses worry or fear |
-| Anger/Frustration | Patient shows frustration or anger |
-| Sadness/Helplessness | Patient feels sad or helpless |
-| Confusion/Doubt | Patient expresses confusion or doubt |
-| Gratitude/Relief | Patient expresses gratitude |
+| 😐 **Neutral** | Objective, non-emotional statements |
+| 😰 **Anxiety/Fear** | Patient expresses worry, concern, or fear |
+| 😠 **Anger/Frustration** | Patient shows frustration or displeasure |
+| 😢 **Sadness/Helplessness** | Patient feels down or hopeless |
+| 🤔 **Confusion/Doubt** | Patient expresses uncertainty or questions |
+| 🙏 **Gratitude/Relief** | Patient conveys thanks or relief |
 
-## Model Performance
+## 📊 Model Performance
 
 ### Overall Metrics
 
 | Metric | Value |
 |--------|-------|
-| Accuracy | 71.3% |
-| Macro F1 | 0.72 |
+| **Accuracy** | **71.3%** |
+| **Macro F1** | **0.722** |
 | Weighted F1 | 0.72 |
 
 ### Per-Class Performance
@@ -61,53 +90,58 @@ NLP emotion classification model for patient-doctor conversations.
 
 ![Label Distribution](data/label_distribution.png)
 
-## Quick Start
+## 🚀 Quick Start
 
 ### 1. Install Dependencies
 
 ```bash
 pip install -r requirements.txt
-```
 
-### 2. Run the Service
+### 2. Launch the Service
 
 ```bash
 cd see
 python app.py
-```
 
-### 3. Open in Browser
+### 3. Access the Interface
 
-```
 http://localhost:8002
-```
 
-## Dataset
+## 📚 Dataset
 
-- **Original Source**: [Chinese MedDialog Dataset](https://tianchi.aliyun.com/dataset/92110) (阿里云天池)
-- **Processing**: Filtered and labeled for 6 emotion categories
-- **Total samples**: 28,280
+This model was trained on a meticulously curated subset of medical dialogues:
+
+- **Original Source**: [Chinese MedDialog Dataset](https://tianchi.aliyun.com/dataset/92110) — Alibaba Cloud Tianchi
+- **Post-Processing**: Carefully filtered, translated, and annotated for emotion classification
+- **Total Samples**: 28,280 annotated dialogues
 - **Categories**: 6 emotion labels
 - **Language**: English
 
-## Model
+## 🏗️ Technical Specifications
 
-- **Base model**: [DistilBERT](https://huggingface.co/distilbert/distilbert-base-uncased)
-- **Task**: 6-class emotion classification
+| Component | Details |
+|-----------|---------|
+| **Base Architecture** | [DistilBERT](https://huggingface.co/distilbert/distilbert-base-uncased) |
+| **Task Type** | 6-class emotion classification |
+| **Max Sequence Length** | 512 tokens |
+| **Framework** | PyTorch + Transformers |
 
-## Directory Structure
+## 📁 Project Structure
 
-```
 patient-emotion-analysis/
-├── best_model/          # Trained model files
+├── best_model/          # Fine-tuned model weights
 ├── see/                 # Inference service
-│   ├── app.py
-│   ├── inference.py
-│   └── templates/
-├── data/                # Training dataset
-├── requirements.txt
-└── README.md
-```
+│   ├── app.py          # Web application
+│   ├── inference.py    # Core inference logic
+│   └── templates/      # UI templates
+├── data/               # Training & evaluation data
+├── requirements.txt    # Dependencies
+└── README.md           # This file
+
+---
 
+<div align="center">
 
+**Blended AI+X Initiative** — *Advancing Healthcare Through Intelligence*
 
+</div>