SidraTul commited on
Commit
f8fa7ea
·
verified ·
1 Parent(s): f0387b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="mental-health-roberta")
53
 
54
  # ✅ Load Mental Disorder Detection Model
55
- mental_bert_pipeline = pipeline("text-classification", model="mental-bert")
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")