Update app.py

app.py

@@ -5,6 +5,9 @@ from transformers import pipeline
 # Create an image classification pipeline
 pipe = pipeline("image-classification", model="trpakov/vit-face-expression")
 
+# Define emotion labels
+emotion_labels = ["Happy", "Sad", "Angry", "Surprised", "Neutral"]
+
 # Streamlit app
 st.title("Emotion Recognition with vit-face-expression")
 
@@ -24,8 +27,7 @@ if uploaded_image:
 
     st.image(image, caption=f"Predicted emotion: {predicted_emotion}", use_column_width=True)
 
-
     # Display scores for each category
     st.write("Emotion Scores:")
-    for label, score in zip(emotion_labels, outputs.logits[0]):
+    for label, score in zip(emotion_labels, results[0]["scores"]):
         st.write(f"{label}: {score:.4f}")