siamese network first push

app/Hackathon_setup/face_recognition.py

@@ -5,6 +5,7 @@ import torch
 import torch.nn.functional as F
 # In the below line,remove '.' while working on your local system. However Make sure that '.' is present before face_recognition_model while uploading to the server, Do not remove it.
 from .face_recognition_model import *
+from .face_recognition_model import Siamese
 from PIL import Image
 import base64
 import io
@@ -74,7 +75,7 @@ def get_similarity(img1, img2):
     ##########################################################################################
     
     # YOUR CODE HERE, load the model
-    myModel = SiameseNetwork().cuda()
+    myModel = Siamese().cuda()
     ckpt = torch.load('./siamese_model.t7')
     myModel.load_state_dict(ckpt['net_dict'])
     myModel.eval()