Initial Comit

app/Hackathon_setup/face_recognition.py

@@ -13,6 +13,7 @@ import pickle
 # Add more imports if required
 from torch.autograd import Variable
 import torch.nn.functional as F
+import joblib
 
 
 ###########################################################################################################################################
@@ -113,6 +114,6 @@ def get_face_class(img1):
   classifier = joblib.load(current_path + '/classifier.pkl.pkl')
   pred = classifier.predict(output).item()
   name = classes[pred]
-  text = pred + " - " + name
+  text = str(pred) + " - " + name
 
   return text