Update app.py

app.py

@@ -13,15 +13,28 @@ def save_uploaded_file(uploaded_file):
         f.write(uploaded_file.getbuffer())
     return file_path
 
+
+
+
 def emotion(file_path):
     if file_path:
         # Classify the file
         out_prob, score, index, text_lab = classifier.classify_file(file_path)
-        # Display the output
-        st.write(text_lab)
+        # Map the original labels to the desired categories
+        emotion_mapping = {
+            'neu': 'Neutral',
+            'ang': 'Angry',
+            'hap': 'Happy',
+            'sad': 'Sadness'
+        }
+        # Get the corresponding category from the mapping
+        emotion_category = emotion_mapping.get(text_lab, 'Unknown')
+        # Return the emotion category
+        st.write(emotion_category)
     else:
         st.write("Please provide the path to an audio file.")
 
+
 def main():
     st.title("Emotion Recognition")
     file_path = st.text_input("Enter the path of the audio file (e.g., /path/to/audio.wav):")