Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -46,7 +46,7 @@ import streamlit as st
|
|
46 |
from transformers import pipeline, AutoTokenizer
|
47 |
|
48 |
# ✅ Load Emotion Recognition Model
|
49 |
-
emotion_pipeline = pipeline("text-classification", model="distilbert-base-uncased-finetuned-emotion")
|
50 |
|
51 |
# ✅ Load Stress Detection Model
|
52 |
stress_pipeline = pipeline("text-classification", model="mental-health-roberta")
|
|
|
46 |
from transformers import pipeline, AutoTokenizer
|
47 |
|
48 |
# ✅ Load Emotion Recognition Model
|
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")
|