Spaces:
Sleeping
Sleeping
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
|
| 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 = ['
|
|
|
|
| 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']
|