updated

app/Hackathon_setup/exp_recognition.py

@@ -69,10 +69,13 @@ def get_expression(img):
 
     if face == 0:
         face = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
-    else:
-        face = torch.from_numpy(np.array(face)).to(device)  # Convert to a PyTorch tensor and move it to the correct device
-
+   
+    with torch.no_grad():
     output = face_det_net(face)
+    _, predicted = torch.max(output, 1)
+
+    predicted_expression = classes[predicted.item()]
+
     
     # YOUR CODE HERE, return expression using your model