Update app.py

app.py

@@ -24,12 +24,10 @@ if uploaded_images:
 
 # Collect selected images based on checkbox input
 selected_images = []
-selected_image_names = []
 for img in uploaded_images:
     selected = st.sidebar.checkbox(f"Select {img.name}", value=False, key=img.name)
     if selected:
         selected_images.append(Image.open(img))
-        selected_image_names.append(img.name)
 
 if st.button("Predict Emotions") and selected_images:
     if len(selected_images) == 2:
@@ -44,7 +42,7 @@ if st.button("Predict Emotions") and selected_images:
             col = col1 if i == 0 else col2
             col.image(selected_images[i], caption=f"Predicted emotion: {predicted_emotion}", use_column_width=True)
             col.write(f"Emotion Scores: {predicted_emotion}: {results[i][0]['score']:.4f}")
-            col.write(f"Original File Name: {selected_image_names[i]}")  # Display original file name
+            col.write(f"Original File Name: {uploaded_images[i].name}")  # Display original file name
 
         # Display the keys and values of all results
         st.write("Keys and Values of all results:")
@@ -68,4 +66,4 @@ if st.button("Predict Emotions") and selected_images:
             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: {selected_image_names[i]}")  # Display original file name
+            st.write(f"Original File Name: {uploaded_images[i].name}")  # Display original file name