Spaces:
Runtime error
Runtime error
CharimCibersegurata
commited on
Commit
•
f33282f
1
Parent(s):
82bb7cc
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ emotion_model_path = 'emotion_detection_model_state.pth'
|
|
15 |
# loading models
|
16 |
face_detection = cv2.CascadeClassifier(detection_model_path)
|
17 |
#load_model(emotion_model_path, compile=False)
|
18 |
-
emotion_classifier = torch.load(emotion_model_path)
|
19 |
EMOTIONS = ["angry", "disgusted", "scared", "happy", "sad", "surprised",
|
20 |
"neutral"]
|
21 |
|
|
|
15 |
# loading models
|
16 |
face_detection = cv2.CascadeClassifier(detection_model_path)
|
17 |
#load_model(emotion_model_path, compile=False)
|
18 |
+
emotion_classifier = torch.load(emotion_model_path,map_location=torch.device('cpu'))
|
19 |
EMOTIONS = ["angry", "disgusted", "scared", "happy", "sad", "surprised",
|
20 |
"neutral"]
|
21 |
|