face recognition

app/Hackathon_setup/face_recognition.py

@@ -136,14 +136,14 @@ def get_face_class(img1):
         embedding = embedding.view(embedding.size(0), -1).cpu().numpy()  # shape (1, embedding_dim)
 
     # 6 Predict class using Decision Tree
-    # pred_label = clf.predict(scaler.transform(embedding))[0]
+    pred_label = clf.predict(scaler.transform(embedding))[0]
 
 
     # --- Predict ---
-    scaled_emb = scaler.transform(embedding)
-    probs = clf.predict_proba(scaled_emb)
-    pred_label = np.argmax(probs)
-    confidence = probs[0, pred_label]
+    # scaled_emb = scaler.transform(embedding)
+    # probs = clf.predict_proba(scaled_emb)
+    # pred_label = np.argmax(probs)
+    # confidence = probs[0, pred_label]
 
 
 
@@ -154,4 +154,4 @@ def get_face_class(img1):
     # class_names = ['Aayush', 'Aditya', 'Vikram']
     # return class_names[pred_label] + " " + str(pred_label)
     class_names = ['Aayush', 'Aditya', 'Vikram']
-    return f"{class_names[pred_label]} ({confidence:.2f})"
+    return f"{class_names[pred_label]} ({pred_label})"