Hugging Face's logo

Spaces:
vrvundyala
/
Siamese_network_hackathon
Sleeping

App Files Community
vrvundyala commited on
Commit
0679ced
·
1 Parent(s): efae1c1

face recognition

Browse files
Files changed (2) hide show
  1. app/Hackathon_setup/decision_tree_model.sav +0 -0
  2. app/Hackathon_setup/face_recognition.py +2 -2
app/Hackathon_setup/decision_tree_model.sav CHANGED
Binary files a/app/Hackathon_setup/decision_tree_model.sav and b/app/Hackathon_setup/decision_tree_model.sav differ
 
app/Hackathon_setup/face_recognition.py CHANGED
@@ -138,5 +138,5 @@ def get_face_class(img1):
138
  # If you have the dataset available:
139
  # class_names = finalClassifierDset.classes
140
  # return class_names[pred_label]
141
- class_names = ['Vikram', 'Vikram', 'Vikram']
142
- return pred_label
 
138
  # If you have the dataset available:
139
  # class_names = finalClassifierDset.classes
140
  # return class_names[pred_label]
141
+ class_names = ['Aayush', 'Aditya', 'Vikram']
142
+ return class_names[pred_label]