Update app.py

app.py

@@ -166,13 +166,13 @@ if st.button("Generate DataFrame") and selected_images:
 
         # Append image metadata and emotion scores to the list
         df_data.append({
-            "Neutral": f"{emotion_scores.get('Neutral', 0.0):.4f}",
-            "Happy": f"{emotion_scores.get('Happy', 0.0):.4f}",
-            "Sad": f"{emotion_scores.get('Sad', 0.0):.4f}",
-            "Angry": f"{emotion_scores.get('Angry', 0.0):.4f}",
-            "Disgust": f"{emotion_scores.get('Disgust', 0.0):.4f}",
-            "Surprise": f"{emotion_scores.get('Surprise', 0.0):.4f}",
-            "Fear": f"{emotion_scores.get('Fear', 0.0):.4f}",  # Add this line if 'Fear' is a possible label
+            "Neutral": f"{emotion_scores.get('neutral', 0.0):.4f}",
+            "Happy": f"{emotion_scores.get('happy', 0.0):.4f}",
+            "Sad": f"{emotion_scores.get('sad', 0.0):.4f}",
+            "Angry": f"{emotion_scores.get('angry', 0.0):.4f}",
+            "Disgust": f"{emotion_scores.get('disgust', 0.0):.4f}",
+            "Surprise": f"{emotion_scores.get('surprise', 0.0):.4f}",
+            "Fear": f"{emotion_scores.get('fear', 0.0):.4f}",  # Add this line if 'Fear' is a possible label
             "File Name": file_name,
             "Size (KB)": size_kb,
             "Timestamp": timestamp.strftime('%Y-%m-%d %H:%M:%S'),  # Format timestamp