vrvundyala commited on
Commit
d6aa61e
·
1 Parent(s): c8432a2

face recognition

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -154,4 +154,4 @@ def get_face_class(img1):
154
  # class_names = ['Aayush', 'Aditya', 'Vikram']
155
  # return class_names[pred_label] + " " + str(pred_label)
156
  class_names = ['Aayush', 'Aditya', 'Vikram']
157
- return f"{class_names[pred_label]} ({pred_label})"
 
154
  # class_names = ['Aayush', 'Aditya', 'Vikram']
155
  # return class_names[pred_label] + " " + str(pred_label)
156
  class_names = ['Aayush', 'Aditya', 'Vikram']
157
+ return f"{class_names[pred_label]} {pred_label}"