Update app.py

app.py

@@ -33,13 +33,16 @@ if st.button("Predict Emotions") and uploaded_images:
             col.image(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: {file_names[i]}")  # Display original file name
+            # Display thumbnail view
+            col.write(f"Thumbnail:")
+            col.image(images[i], use_column_width='auto')
 
         # Display the keys and values of all results
-        st.write("Emotion Score Summary :")
+        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 col2
-            col.write(f"results for #[{i+1}] image")
+            col.write(f"Keys and Values of results[{i}]:")
             for res in result:
                 label = res["label"]
                 score = res["score"]
@@ -60,3 +63,6 @@ if st.button("Predict Emotions") and uploaded_images:
             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
+            # Display thumbnail view
+            st.write(f"Thumbnail:")
+            st.image(images[i], use_column_width='auto')