Update app.py

app.py

@@ -51,4 +51,22 @@ if st.button("Predict Emotions") and selected_images:
         st.write("Keys and Values of all results:")
         col1, col2 = st.columns(2)
         for i, result in enumerate(results):
-            col = col1 if i == 0 else 
+            col = col1 if i == 0 else col2
+            col.write(f"Keys and Values of results[{i}]:")
+            for res in result:
+                label = res["label"]
+                score = res["score"]
+                col.write(f"{label}: {score:.4f}")
+            
+    else:
+        # Predict emotion for each selected image using the pipeline
+        results = [pipe(image) for image in selected_images]
+
+        # Display images and predicted emotions
+        for i, result in enumerate(results):
+            predicted_class = result[0]["label"]
+            predicted_emotion = predicted_class.split("_")[-1].capitalize()
+            st.image(selected_images[i], caption=f"Predicted emotion: {predicted_emotion}", use_column_width=True)
+            st.write(f"Emotion Scores for #{i+1} Image")
+            st.write(f"{predicted_emotion}: {result[0]['score']:.4f}")
+            st.write(f"Original File Name: {uploaded_images[i].name}")  # Display original file name