manikanta2026 commited on
Commit
3f0edb4
·
1 Parent(s): 3ed80eb
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -45,9 +45,9 @@ def predict_emotion(audio_file):
45
  predicted_class = np.argmax(predictions[0])
46
  predicted_emotion = label_encoder.inverse_transform([predicted_class])[0]
47
 
48
- # Output confidences as floats (0-100), rounded to 2 decimals
49
  emotion_probabilities = {
50
- label_encoder.inverse_transform([i])[0]: round(float(pred) * 100, 2)
51
  for i, pred in enumerate(predictions[0])
52
  }
53
 
 
45
  predicted_class = np.argmax(predictions[0])
46
  predicted_emotion = label_encoder.inverse_transform([predicted_class])[0]
47
 
48
+ # Output confidences as floats (0-1), rounded to 4 decimals
49
  emotion_probabilities = {
50
+ label_encoder.inverse_transform([i])[0]: round(float(pred), 4)
51
  for i, pred in enumerate(predictions[0])
52
  }
53