Spaces:
Sleeping
Sleeping
Pallavi Bhoj
commited on
Commit
·
e1575d9
1
Parent(s):
5e9d984
Update face_recognition.py
Browse files
app/Hackathon_setup/face_recognition.py
CHANGED
@@ -77,7 +77,7 @@ def get_similarity(img1, img2):
|
|
77 |
|
78 |
|
79 |
|
80 |
-
output1, output2 = feature_net(
|
81 |
euclidean_distance = F.pairwise_distance(output1, output2)
|
82 |
|
83 |
|
|
|
77 |
|
78 |
|
79 |
|
80 |
+
output1, output2 = feature_net((face1).to(device), (face2).to(device))
|
81 |
euclidean_distance = F.pairwise_distance(output1, output2)
|
82 |
|
83 |
|