Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
|
|
4 |
|
5 |
# 1. Emotion Detection Model (Using Hugging Face's transformer)
|
6 |
# Choose a suitable model - 'emotion-classification' is the task, you can specify a model from Hugging Face Model Hub.
|
7 |
-
emotion_classifier = pipeline("text-classification", model="
|
8 |
|
9 |
# 2. Conversational Agent Logic
|
10 |
def get_ai_response(user_input, emotion_predictions):
|
|
|
4 |
|
5 |
# 1. Emotion Detection Model (Using Hugging Face's transformer)
|
6 |
# Choose a suitable model - 'emotion-classification' is the task, you can specify a model from Hugging Face Model Hub.
|
7 |
+
emotion_classifier = pipeline("text-classification", model="AnasAlokla/multilingual_go_emotions") # Or choose another model
|
8 |
|
9 |
# 2. Conversational Agent Logic
|
10 |
def get_ai_response(user_input, emotion_predictions):
|