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 = ["anger", "happy", "sad", "afraid", "worried"]
+    labels = ["anger", "Joy", "Sadness", "Fear", "Neutral", "Surprise"]
     results = {label: prob for label, prob in zip(labels, probs)}
     return results