Update README.md

README.md

@@ -14,8 +14,17 @@ tags:
 - moe
 ---
 
-CSV_FILES = [
-    ("results-diagnose.csv", 0),  # path, label
-    ("results-treatment.csv", 1),
-    ("results-psy.csv", 2),
-]
+# MoE Router Model
+
+Classify clinical text into:
+
+* **0:** Diagnosis
+* **1:** Treatment
+* **2:** Psychological Support
+
+## Training
+
+* **Base model:** ModernBERT-base
+* **Epochs:** 3
+* **Learning rate:** 3e-5
+* **Batch size:** 16