Spaces:

xtlyxt
/

facefeature

Sleeping

xtlyxt commited on Apr 26, 2024

Commit

5d8bb70

verified ·

1 Parent(s): 8552902

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch
 pipe = pipeline("image-classification", model="trpakov/vit-face-expression", top_k=None)
 # Define emotion labels
-emotion_labels = ["Happy", "Sad", "Angry", "Surprised", "Neutral"]
 # Streamlit app
 st.title("Emotion Recognition with vit-face-expression")

 pipe = pipeline("image-classification", model="trpakov/vit-face-expression", top_k=None)
 # Define emotion labels
+emotion_labels = ["Neutral", "Sad", "Angry", "Surprised", "Happy"]
 # Streamlit app
 st.title("Emotion Recognition with vit-face-expression")