Spaces:
Runtime error
Runtime error
Dhrumit1314
commited on
Commit
•
ffe8de0
1
Parent(s):
78fcc66
Update app.py
Browse files
app.py
CHANGED
@@ -13,10 +13,10 @@ import numpy as np
|
|
13 |
import datetime
|
14 |
|
15 |
# Load models
|
16 |
-
face_classifier = cv2.CascadeClassifier('
|
17 |
-
emotion_model = load_model('
|
18 |
-
age_model = load_model('
|
19 |
-
gender_model = load_model('
|
20 |
|
21 |
class_labels = ['Angry', 'Disgust', 'Fear', 'Happy', 'Neutral', 'Sad', 'Surprise']
|
22 |
gender_labels = ['Male', 'Female']
|
|
|
13 |
import datetime
|
14 |
|
15 |
# Load models
|
16 |
+
face_classifier = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
|
17 |
+
emotion_model = load_model('emotion_detection_model_50epochs.h5')
|
18 |
+
age_model = load_model('age_model_3epochs.h5')
|
19 |
+
gender_model = load_model('gender_model_3epochs.h5')
|
20 |
|
21 |
class_labels = ['Angry', 'Disgust', 'Fear', 'Happy', 'Neutral', 'Sad', 'Surprise']
|
22 |
gender_labels = ['Male', 'Female']
|