siamese network first push

app/Hackathon_setup/face_recognition.py

@@ -79,7 +79,7 @@ def get_similarity(img1, img2):
     myModel = Siamese().to(device)
     BASE_DIR = os.path.dirname(os.path.abspath(__file__))
     ckpt_path = os.path.join(BASE_DIR, "siamese_model.t7")
-    ckpt = torch.load(ckpt_path)
+    ckpt = torch.load(ckpt_path, map_location=device)
     myModel.load_state_dict(ckpt['net_dict'])
     myModel.eval()