Pallavi Bhoj commited on
Commit
5e9d984
·
1 Parent(s): 434cd9b

Update face_recognition.py

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -72,7 +72,7 @@ def get_similarity(img1, img2):
72
  ##########################################################################################
73
  ##########################################################################################
74
  feature_net = Siamese()
75
- model = torch.load('./siamese_model.t7', map_location=device)
76
  feature_net.load_state_dict(model['net_dict'])
77
 
78
 
 
72
  ##########################################################################################
73
  ##########################################################################################
74
  feature_net = Siamese()
75
+ model = torch.load(current_path + '/siamese_model.t7', map_location=device)
76
  feature_net.load_state_dict(model['net_dict'])
77
 
78