Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import pipeline
 pipe = pipeline("image-classification", model="trpakov/vit-face-expression", top_k=None)
 
 # Define emotion labels
-emotion_labels = ["Neutral", "Sad", "Angry", "Surprised", "Happy"]
+#emotion_labels = ["Neutral", "Sad", "Angry", "Surprised", "Happy"]
 
 # Streamlit app
 st.title("Emotion Recognition with vit-face-expression")
@@ -48,6 +48,6 @@ if st.button("Predict Emotions") and uploaded_images:
             predicted_class = result[0]["label"]
             predicted_emotion = predicted_class.split("_")[-1].capitalize()
             st.image(images[i], caption=f"Predicted emotion: {predicted_emotion}", use_column_width=True)
-            st.write(f"Emotion Scores for #{i+1}: Image")
+            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