Update app.py

app.py

@@ -156,8 +156,7 @@ def predict_emotion(context):
     emotion_prediction_pipeline = pipeline('text-classification', model=emotion_prediction_model, tokenizer=emotion_prediction_tokenizer, top_k=None)
     predictions = emotion_prediction_pipeline(context)
     emotion_scores = predictions[0]
-    sorted_scores = sorted(emotion_scores.items(), key=lambda x: x[1], reverse=True)
-    emotion_pred = sorted_scores[0][0]
+    emotion_pred = emotion_classes[np.argmax(emotion_scores)]
     return emotion_pred
 
 def generate_text(prompt, max_length=100, emotion=None):