changes

app/Hackathon_setup/face_recognition.py

@@ -111,4 +111,15 @@ def get_face_class(img1):
     ##YOUR CODE HERE, return face class here
     ##Hint: you need a classifier finetuned for your classes, it takes o/p of siamese as i/p to it
     ##Better Hint: Siamese experiment is covered in one of the labs
-    return "YET TO BE CODED"
+
+    feature_net = SiameseNetwork() #Example Network 
+    feature_net = feature_net.to(device)
+    current_path = 'app/Hackathon_setup'
+    model = torch.load(current_path + '/siamese_model.t7', map_location=device)
+    feature_net.load_state_dict(model['net_dict'])                      
+    
+    output1 = feature_net.forward_once(det_img1.to(device))
+    probabilities = torch.softmax(outputs, dim=1)
+    predicted_class = torch.argmax(probabilities, dim=1).item()
+    
+    return predicted_class