manikanta2026
commited on
Commit
·
3f0edb4
1
Parent(s):
3ed80eb
changes3
Browse files
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-
|
49 |
emotion_probabilities = {
|
50 |
-
label_encoder.inverse_transform([i])[0]: round(float(pred)
|
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 |
|