Spaces:

sid-reddy-krishna
/

aimliiith

Sleeping

sid-reddy-krishna commited on Oct 29, 2023

Commit

2fd8c7c

1 Parent(s): 87fac06

exp rec code

Files changed (1) hide show

app/Hackathon_setup/exp_recognition.py CHANGED Viewed

@@ -74,6 +74,6 @@ def get_expression(img):
     face = trnscm(face).unsqueeze(0)
     output = face_det_net(face)
     _, pred = torch.max(output, dim=1)
-    return classes[pred]
     return "YET TO BE CODED"

     face = trnscm(face).unsqueeze(0)
     output = face_det_net(face)
     _, pred = torch.max(output, dim=1)
+    return classes[pred.item()]
     return "YET TO BE CODED"