files added

app/Hackathon_setup/face_recognition.py

@@ -3,7 +3,7 @@ import cv2
 from matplotlib import pyplot as plt
 import torch
 # 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 . import face_recognition_model
+from .face_recognition_model import *
 from PIL import Image
 import base64
 import io
@@ -74,7 +74,7 @@ def get_similarity(img1, img2):
     ##########################################################################################
     
     # YOUR CODE HERE, load the model
-    feature_net = face_recognition_model.Siamese()                                   
+    feature_net = Siamese()                                   
     model = torch.load( './siamese_model.t7', map_location=device)          
     feature_net.load_state_dict(model['net_dict'])