Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -49,10 +49,10 @@ from transformers import pipeline, AutoTokenizer
|
|
49 |
emotion_pipeline = pipeline("text-classification", model="ahmettasdemir/distilbert-base-uncased-finetuned-emotion")
|
50 |
|
51 |
# ✅ Load Stress Detection Model
|
52 |
-
stress_pipeline = pipeline("text-classification", model="
|
53 |
|
54 |
# ✅ Load Mental Disorder Detection Model
|
55 |
-
mental_bert_pipeline = pipeline("text-classification", model="
|
56 |
|
57 |
# ✅ Load PHQ-9 Depression Severity Classifier
|
58 |
phq9_pipeline = pipeline("text-classification", model="PHQ-9 Depression Classifier")
|
|
|
49 |
emotion_pipeline = pipeline("text-classification", model="ahmettasdemir/distilbert-base-uncased-finetuned-emotion")
|
50 |
|
51 |
# ✅ Load Stress Detection Model
|
52 |
+
stress_pipeline = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")
|
53 |
|
54 |
# ✅ Load Mental Disorder Detection Model
|
55 |
+
mental_bert_pipeline = pipeline("text-classification", model="nlpconnect/vit-gpt2-image-captioning")
|
56 |
|
57 |
# ✅ Load PHQ-9 Depression Severity Classifier
|
58 |
phq9_pipeline = pipeline("text-classification", model="PHQ-9 Depression Classifier")
|