device changes

app/Hackathon_setup/face_recognition.py

@@ -85,12 +85,13 @@ def get_similarity(img1, img2):
     ##########################################################################################
     
     # YOUR CODE HERE, load the model
-    feature_net = SiameseNetwork().cuda() #Example Network                                           ##
-    model = torch.load(current_path + '/siamese_model.t7', map_location=device)          ##
+    feature_net = SiameseNetwork() #Example Network 
+    feature_net = feature_net.to(device)
+    model = torch.load('/siamese_model.t7', map_location=device)
     feature_net.load_state_dict(model['net_dict'])                      
     
     # YOUR CODE HERE, return similarity measure using your model
-    output1,output2 = feature_net(face1.cuda(),face2.cuda())
+    output1,output2 = feature_net(face1.to(device),face2.to(device))
     euclidean_distance = F.pairwise_distance(output1, output2)
     return euclidean_distance