Spaces:
Sleeping
Sleeping
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 =
|
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 |
|