Spaces:
Runtime error
Runtime error
Update app/Hackathon_setup/exp_recognition.py
Browse files
app/Hackathon_setup/exp_recognition.py
CHANGED
|
@@ -70,6 +70,7 @@ def get_expression(img):
|
|
| 70 |
face_det_net = facExpRec()
|
| 71 |
model = torch.load(current_path + '/exp_recognition_net.t7', map_location=device)
|
| 72 |
face_det_net.load_state_dict(model['net_dict'])
|
|
|
|
| 73 |
face = trnscm(face).unsqueeze(0)
|
| 74 |
print(face.shape)
|
| 75 |
output = face_det_net(face)
|
|
|
|
| 70 |
face_det_net = facExpRec()
|
| 71 |
model = torch.load(current_path + '/exp_recognition_net.t7', map_location=device)
|
| 72 |
face_det_net.load_state_dict(model['net_dict'])
|
| 73 |
+
face_det_net.eval()
|
| 74 |
face = trnscm(face).unsqueeze(0)
|
| 75 |
print(face.shape)
|
| 76 |
output = face_det_net(face)
|