Update app.py

app.py

@@ -40,3 +40,9 @@ if uploaded_image:
     predicted_emotion = emotion_labels[predicted_class]
 
     st.image(image, caption=f"Predicted emotion: {predicted_emotion}", use_column_width=True)
+
+
+    # Display scores for each category
+    st.write("Emotion Scores:")
+    for label, score in zip(emotion_labels, outputs.logits[0]):
+        st.write(f"{label}: {score:.4f}")