similarity code is updated01

app/Hackathon_setup/face_recognition.py

@@ -2,7 +2,7 @@ import numpy as np
 import cv2
 from matplotlib import pyplot as plt
 import torch
-
+from torch.autograd import Variable
 # 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 PIL import Image
@@ -91,7 +91,8 @@ def get_similarity(img1, img2):
 
     # YOUR CODE HERE, return similarity measure using your model
     feature_net.eval()
-    output1,output2 = feature_net(face1,face2)
+
+    output1,output2 = feature_net(Variable(face1),Variable(face2))
     euclidean_distance = F.pairwise_distance(output1, output2)
     #features_1 = feature_net(face1)[1]
     #features_2 = feature_net(face2)[1]