safwansajad commited on
Commit
21c696d
·
verified ·
1 Parent(s): 56226b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ tokenizer = AutoTokenizer.from_pretrained(chatbot_model)
7
  model = AutoModelForCausalLM.from_pretrained(chatbot_model)
8
 
9
  # Load emotion detection model
10
- emotion_pipeline = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")
11
 
12
  st.title("🧠 Mental Health Chatbot")
13
 
 
7
  model = AutoModelForCausalLM.from_pretrained(chatbot_model)
8
 
9
  # Load emotion detection model
10
+ emotion_pipeline = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion")
11
 
12
  st.title("🧠 Mental Health Chatbot")
13