Dpngtm commited on
Commit
1f55dee
·
verified ·
1 Parent(s): 9caa94e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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