Spaces:

tsaddev
/

facerecognization

Runtime error

tsaddev commited on Oct 29, 2023

Commit

2431053

1 Parent(s): 3bd78ba

Update app/Hackathon_setup/exp_recognition.py

Files changed (1) hide show

app/Hackathon_setup/exp_recognition.py CHANGED Viewed

@@ -76,4 +76,4 @@ def get_expression(img):
     output = face_det_net(face)
     print(output.shape)
     _, pred = torch.max(output, dim=1)
-    return classes[pred]

     output = face_det_net(face)
     print(output.shape)
     _, pred = torch.max(output, dim=1)
+    return classes[pred.item()]