Update app.py

app.py

@@ -31,11 +31,8 @@ if st.button("Predict Emotions") and uploaded_images:
         predicted_class = result[0]["label"]
         predicted_emotion = predicted_class.split("_")[-1].capitalize()
         st.image(images[i], caption=f"Predicted emotion: {predicted_emotion}", use_column_width=True)
-
-        # Display scores for all categories
         st.write(f"Emotion Scores for Image {i+1}:")
-        for emotion, score in result[0]["score"].items():
-            st.write(f"{emotion.capitalize()}: {score:.4f}")
+        st.write(f"{predicted_emotion}: {result[0]['score']:.4f}")
 
         # If only two images and not the last image, display side by side
         if num_images == 2 and i == 0: