Spaces:

sid-reddy-krishna
/

aimliiith

Sleeping

sid-reddy-krishna commited on Oct 29, 2023

Commit

eaab397

1 Parent(s): a782c75

fix

Files changed (1) hide show

app/Hackathon_setup/face_recognition.py CHANGED Viewed

@@ -104,7 +104,7 @@ def get_face_class(img1):
     ##Better Hint: Siamese experiment is covered in one of the labs
     classifier_net = Classifier()
     model = torch.load(current_path + '/classifier_model.t7',map_location=torch.device('cpu'))
-    feature_net.load_state_dict(model['net_dict'])
     feature_net = SiameseNetwork()
     model = torch.load(current_path + '/siamese_model.t7',map_location=torch.device('cpu'))

     ##Better Hint: Siamese experiment is covered in one of the labs
     classifier_net = Classifier()
     model = torch.load(current_path + '/classifier_model.t7',map_location=torch.device('cpu'))
+    classifier_net.load_state_dict(model['net_dict'])
     feature_net = SiameseNetwork()
     model = torch.load(current_path + '/siamese_model.t7',map_location=torch.device('cpu'))