Update app.py

app.py

@@ -14,7 +14,7 @@ def classify_emotion(text):
     logits = outputs.logits
     probabilities = torch.nn.functional.softmax(logits, dim=-1)
     probs = probabilities.detach().numpy()[0]
-    labels = ["Angry", "Happy", "Sad", "Afraid", "Surprised"]
+    labels = ["anger", "disgust", "fear", "joy", "neutral", "sadness", "surprise"]
     results = {label: prob for label, prob in zip(labels, probs)}
     return results