Spaces:

tsaddev
/

facerecognization

Runtime error

tsaddev commited on Oct 28, 2023

Commit

1209bf8

1 Parent(s): 831f4bb

Update app/Hackathon_setup/face_recognition.py

Files changed (1) hide show

app/Hackathon_setup/face_recognition.py CHANGED Viewed

@@ -78,7 +78,8 @@ def get_similarity(img1, img2):
     siameseNetwork.load_state_dict(model['net_dict'])
     # YOUR CODE HERE, return similarity measure using your model
-    euclidean_distance = F.pairwise_distance(img1, img2)
     return euclidean_distance

     siameseNetwork.load_state_dict(model['net_dict'])
     # YOUR CODE HERE, return similarity measure using your model
+    out1, out2 = siameseNetwork(img1, img2)
+    euclidean_distance = F.pairwise_distance(out1, out2)
     return euclidean_distance