Update app/Hackathon_setup/face_recognition.py

app/Hackathon_setup/face_recognition.py

@@ -52,7 +52,8 @@ def detected_face(image):
 #5) For loading your model use the current_path+'your model file name', anyhow detailed example is given in comments to the function 
 #Caution: Don't change the definition or function name; for loading the model use the current_path for path example is given in comments to the function
 def get_similarity(img1, img2):
-    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+    # device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+    device = torch.device("cpu")
     
     det_img1 = detected_face(img1)
     det_img2 = detected_face(img2)
@@ -79,7 +80,7 @@ def get_similarity(img1, img2):
     
     # YOUR CODE HERE, return similarity measure using your model
     out1, out2 = siameseNetwork(face1, face2)
-    euclidean_distance = F.pairwise_distance(out1, out2)
+    euclidean_distance = F.pairwise_distance(out1, out2).item()
     
     return euclidean_distance