Fix model path for Docker runtime

model/inference.py

@@ -8,7 +8,13 @@ from PIL import Image
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # load checkpoint
-checkpoint = torch.load("emotion_recognition_model.pth", map_location=device)
+from pathlib import Path
+
+BASE_DIR = Path(__file__).resolve().parent
+MODEL_PATH = BASE_DIR / "emotion_recognition_model.pth"
+
+checkpoint = torch.load(MODEL_PATH, map_location=device)
+
 classes = checkpoint["classes"]
 
 # recreate model