Spaces:

abajaj
/

Face_Similarity_and_Recognition

Runtime error

abajaj commited on Oct 28, 2023

Commit

58c976e

•

1 Parent(s): 9f14dd4

Initial Comit

Files changed (1) hide show

app/Hackathon_setup/exp_recognition.py CHANGED Viewed

@@ -75,7 +75,7 @@ def get_expression(img):
   with torch.no_grad():
     output = face_det_net(updatedface)
     _, predicted = torch.max(output, 1)
-    label = classes[predicted]
     expression = str(predicted) + " - " + label
   # YOUR CODE HERE, return expression using your model

   with torch.no_grad():
     output = face_det_net(updatedface)
     _, predicted = torch.max(output, 1)
+    label = classes[predicted.item()]
     expression = str(predicted) + " - " + label
   # YOUR CODE HERE, return expression using your model