Update app.py

app.py

@@ -53,17 +53,16 @@ if st.button("Predict Emotions") and selected_images:
                 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):
+        for i, (image, result) in enumerate(zip(selected_images, 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.image(image, 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}")
             # Use the index to get the corresponding filename
-            st.write(f"Original File Name: {selected_images[i].filename}")  
+            st.write(f"Original File Name: {uploaded_images[i].name if i < len(uploaded_images) else 'Unknown'}")