Update app/Hackathon_setup/exp_recognition.py

app/Hackathon_setup/exp_recognition.py

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