CharimCibersegurata commited on
Commit
f33282f
1 Parent(s): 82bb7cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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