Update app.py

app.py

@@ -20,6 +20,8 @@ def emotion(file_path):
     if file_path:
         # Classify the file
         out_prob, score, index, text_lab = classifier.classify_file(file_path)
+        if isinstance(text_lab, list):
+            text_lab = text_lab[0]
         # Map the original labels to the desired categories
         emotion_mapping = {
             'neu': 'Neutral',
@@ -28,6 +30,8 @@ def emotion(file_path):
             'sad': 'Sadness'
         }
         # Get the corresponding category from the mapping
+        emotion_category = emotion_mapping.get(text_lab, 'Unknown')
+        
         emotion_category = emotion_mapping.get(text_lab, 'Unknown')
         # Return the emotion category
         st.write(emotion_category)