Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,9 +12,7 @@ st.write(f"Using device: {device}")
|
|
| 12 |
|
| 13 |
# Load emotion model
|
| 14 |
learn_emotion = load_learner('emotions_vgg19.pkl').to(device)
|
| 15 |
-
|
| 16 |
-
# Manually define emotion labels
|
| 17 |
-
learn_emotion_labels = ["Happy", "Sad", "Neutral", "Angry", "Disgust", "Fear", "Surprise"]
|
| 18 |
|
| 19 |
# Load Haar Cascade for face detection
|
| 20 |
face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + "haarcascade_frontalface_default.xml")
|
|
|
|
| 12 |
|
| 13 |
# Load emotion model
|
| 14 |
learn_emotion = load_learner('emotions_vgg19.pkl').to(device)
|
| 15 |
+
learn_emotion_labels = learn_emotion.dls.vocab
|
|
|
|
|
|
|
| 16 |
|
| 17 |
# Load Haar Cascade for face detection
|
| 18 |
face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + "haarcascade_frontalface_default.xml")
|