npuliga commited on
Commit
10afa34
·
1 Parent(s): e8d631e

updated class names

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -185,7 +185,7 @@ def get_face_class(img1):
185
  else:
186
  print("ERROR: No classifier file found!")
187
  return "Unknown"
188
- except Exception as classifier_ERROR:
189
  print(f"ERROR: Failed to load classifier: {classifier_error}")
190
  return handle_missing_model("Failed to load classifier")
191
 
 
185
  else:
186
  print("ERROR: No classifier file found!")
187
  return "Unknown"
188
+ except Exception as classifier_error:
189
  print(f"ERROR: Failed to load classifier: {classifier_error}")
190
  return handle_missing_model("Failed to load classifier")
191
 
app/Hackathon_setup/face_recognition_model.py CHANGED
@@ -67,4 +67,4 @@ class ContrastiveLoss(torch.nn.Module):
67
  return loss_contrastive
68
 
69
  # Definition of classes as dictionary - Updated to match 5 classes from training
70
- classes = ['Aditya','Aayush','Vikram','Aditi','Suchitra']
 
67
  return loss_contrastive
68
 
69
  # Definition of classes as dictionary - Updated to match 5 classes from training
70
+ classes = ['Aayush','Aditya','Vikram','Aditi','Suchitra']