Update app/Hackathon_setup/exp_recognition.py

app/Hackathon_setup/exp_recognition.py

@@ -71,6 +71,8 @@ def get_expression(img):
     model = torch.load(current_path + '/exp_recognition_net.t7', map_location=device)
     face_det_net.load_state_dict(model['net_dict'])
     face = trnscm(face).unsqueeze(0)
+    print(face.shape)
     output = face_det_net(face)
-    _, pred = torch.max(output)
+    print(output.shape)
+    _, pred = torch.max(output, dim=1)
     return classes[pred]