Update app.py
Browse files
app.py
CHANGED
|
@@ -40,9 +40,9 @@ class_names = [
|
|
| 40 |
"Bipolar",
|
| 41 |
"Depression",
|
| 42 |
"Normal",
|
| 43 |
-
"
|
| 44 |
"Stress",
|
| 45 |
-
"
|
| 46 |
]
|
| 47 |
|
| 48 |
# ======== Load Tokenizer & Model ========
|
|
|
|
| 40 |
"Bipolar",
|
| 41 |
"Depression",
|
| 42 |
"Normal",
|
| 43 |
+
"Personality disorder",
|
| 44 |
"Stress",
|
| 45 |
+
"Suicidal"
|
| 46 |
]
|
| 47 |
|
| 48 |
# ======== Load Tokenizer & Model ========
|