Spaces:

jeevana
/

group14_hackathon

Sleeping

jeevana commited on Oct 29, 2023

Commit

ba503cb

1 Parent(s): 95bd875

classifier03

Files changed (1) hide show

app/Hackathon_setup/face_recognition.py CHANGED Viewed

@@ -140,7 +140,7 @@ def get_face_class(img1):
     outputs = feature_classifier(representations)
     _, predicted = torch.max(outputs.data, 1)
     ##Better Hint: Siamese experiment is covered in one of the labs
-    return predicted

     outputs = feature_classifier(representations)
     _, predicted = torch.max(outputs.data, 1)
     ##Better Hint: Siamese experiment is covered in one of the labs
+    return predicted.item()