face expressions

app/Hackathon_setup/exp_recognition.py

@@ -69,7 +69,9 @@ def get_expression(img):
     optimizer = torch.optim.Adam(model.parameters(), lr=0.0001)
 
     # Load the checkpoint
-    checkpoint = torch.load('resnet_expression_recognition.pth', map_location=device)
+    BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+    ckpt_path = os.path.join(BASE_DIR, "expression_model.t7")
+    checkpoint = torch.load(ckpt_path, map_location=device)
 
     # Restore weights and optimizer
     model.load_state_dict(checkpoint['model_state_dict'])