Pallavi Bhoj commited on
Commit
4f5950a
·
1 Parent(s): a8714db

Update face_recognition.py

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -71,7 +71,7 @@ def get_similarity(img1, img2):
71
  ##the same path as this file, we recommend to put in the same directory ##
72
  ##########################################################################################
73
  ##########################################################################################
74
- feature_net = face_recognition_model.Siamese()
75
  model = torch.load('./siamese_model.t7', map_location=device)
76
  feature_net.load_state_dict(model['net_dict'])
77
 
 
71
  ##the same path as this file, we recommend to put in the same directory ##
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