AnasAlokla commited on
Commit
8b87e1a
·
verified ·
1 Parent(s): ef12202

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="SamLowe/roberta-base-go_emotions") # Or choose another 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):