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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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")