Spaces:
Sleeping
Sleeping
modify file path
Browse files- prediction.py +2 -2
prediction.py
CHANGED
|
@@ -5,10 +5,10 @@ from keras.models import load_model
|
|
| 5 |
import pandas as pd
|
| 6 |
|
| 7 |
# Load pre-trained emotion classification model
|
| 8 |
-
emotion_classification_model = load_model('.
|
| 9 |
|
| 10 |
# Load Haarcascades face detection classifier
|
| 11 |
-
face_cascade = cv2.CascadeClassifier('.
|
| 12 |
|
| 13 |
class_labels = ["angry", "disgusted", "fearful", "happy", "neutral", "sad", "surprised"]
|
| 14 |
|
|
|
|
| 5 |
import pandas as pd
|
| 6 |
|
| 7 |
# Load pre-trained emotion classification model
|
| 8 |
+
emotion_classification_model = load_model('./model/model_fine_tune.h5') # Replace with actual path
|
| 9 |
|
| 10 |
# Load Haarcascades face detection classifier
|
| 11 |
+
face_cascade = cv2.CascadeClassifier('./model/haarcascade_frontalface_default.xml')
|
| 12 |
|
| 13 |
class_labels = ["angry", "disgusted", "fearful", "happy", "neutral", "sad", "surprised"]
|
| 14 |
|