Spaces:
Sleeping
Sleeping
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('
|
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 |
|