Update app.py

app.py

@@ -20,6 +20,17 @@ id2label = {
     6: "Surprised"
 }
 
+emotion_emojis = {
+    "Neutral": "😐",
+    "Happy": "😊",
+    "Sad": "😢",
+    "Angry": "😠",
+    "Fearful": "😨",
+    "Disgusted": "🤢",
+    "Surprised": "😲"
+}
+
+
 
 # Function to classify emotions from audio
 def classify_emotion(audio_file):
@@ -42,8 +53,10 @@ def classify_emotion(audio_file):
     
     # Convert class ID to emotion label
     predicted_emotion = id2label.get(predicted_class_id, "Unknown")
-    
-    return f"Predicted Emotion: {predicted_emotion}"
+    emoji = emotion_emojis.get(predicted_emotion, "❓")
+
+    return f"Predicted Emotion: {predicted_emotion} {emoji}"
+
 
 # Gradio Interface
 interface = gr.Interface(
@@ -51,7 +64,7 @@ interface = gr.Interface(
     inputs=gr.Audio(type="filepath"),
     outputs="text",
     title="Speech Emotion Classifier 🎭",
-    description="Upload an audio file and the model will classify its emotion (e.g., Happy, Sad, Angry)."
+    description="Upload an audio file and the model will classify its emotion (Neutral, Happy, Sad, Angry, Fearful, Disgust, Surprised)."
 )
 
 # Launch the app