Update app.py
Browse files
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="
|
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 |
|