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")
|