Spaces:

vrvundyala
/

Siamese_network_hackathon

Sleeping

vrvundyala commited on Oct 26, 2025

Commit

50f9fa8

1 Parent(s): 871e118

face recognition

Files changed (2) hide show

app/Hackathon_setup/face_recognition.py CHANGED Viewed

@@ -116,7 +116,7 @@ def get_face_class(img1):
     # 2 Load the Siamese feature extractor
     myModel = Siamese().to(device)
-    ckpt_path = os.path.join(BASE_DIR, "siamese_model.t7")
     ckpt = torch.load(ckpt_path, map_location=device)
     myModel.load_state_dict(ckpt['net_dict'])
     myModel.eval()

     # 2 Load the Siamese feature extractor
     myModel = Siamese().to(device)
+    ckpt_path = os.path.join(BASE_DIR, "siamese_model_1.t7")
     ckpt = torch.load(ckpt_path, map_location=device)
     myModel.load_state_dict(ckpt['net_dict'])
     myModel.eval()

app/Hackathon_setup/siamese_model_1.t7 ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:92b34f44a4f8db131f1cc6b8b2e9e66a8618b0b4b13c92ec8202d57dd88e6759
+size 161027032