Update app.py
Browse files
app.py
CHANGED
@@ -62,9 +62,12 @@ def recognize_emotion(audio):
|
|
62 |
outputs = model(input_values)
|
63 |
logits = outputs.logits
|
64 |
probs = F.softmax(logits, dim=-1)[0].cpu().numpy()
|
|
|
|
|
|
|
65 |
|
66 |
confidence_scores = {
|
67 |
-
f"{emotion} {emotion_icons[emotion]}": round(float(prob),2)
|
68 |
for emotion, prob in zip(emotion_labels, probs)
|
69 |
}
|
70 |
|
|
|
62 |
outputs = model(input_values)
|
63 |
logits = outputs.logits
|
64 |
probs = F.softmax(logits, dim=-1)[0].cpu().numpy()
|
65 |
+
|
66 |
+
# Ensure probabilities sum to 1 and convert to percentages
|
67 |
+
probs = probs / probs.sum() # Normalize to ensure sum is 1
|
68 |
|
69 |
confidence_scores = {
|
70 |
+
f"{emotion} {emotion_icons[emotion]}": round(float(prob)*100,2)
|
71 |
for emotion, prob in zip(emotion_labels, probs)
|
72 |
}
|
73 |
|