siamese network first push

app/Hackathon_setup/face_recognition.py

@@ -77,7 +77,9 @@ def get_similarity(img1, img2):
     # YOUR CODE HERE, load the model
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     myModel = Siamese().to(device)
-    ckpt = torch.load('siamese_model.t7')
+    BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+    ckpt_path = os.path.join(BASE_DIR, "siamese_model.t7")
+    ckpt = torch.load(ckpt_path)
     myModel.load_state_dict(ckpt['net_dict'])
     myModel.eval()