changes

app/Hackathon_setup/face_recognition.py

@@ -118,7 +118,7 @@ def get_face_class(img1):
     model = torch.load(current_path + '/siamese_model_recog.t7', map_location=device)
     feature_net.load_state_dict(model['net_dict'])                      
     
-    output1 = feature_net.forward_once(det_img1.to(device))
+    output1 = feature_net.forward_once(det_img1)
     probabilities = torch.softmax(outputs, dim=1)
     predicted_class = torch.argmax(probabilities, dim=1).item()