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(Variable(face1).to(device), Variable(face2).to(device))
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