Updated inference.py to clear the model not found runtime error

model/inference.py

@@ -3,18 +3,17 @@ import torch
 import torch.nn as nn
 from torchvision import models, transforms
 from PIL import Image
+from pathlib import Path
 
 # device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# load checkpoint
-from pathlib import Path
-
+# resolve model path safely
 BASE_DIR = Path(__file__).resolve().parent
 MODEL_PATH = BASE_DIR / "emotion_recognition_model.pth"
 
+# load checkpoint
 checkpoint = torch.load(MODEL_PATH, map_location=device)
-
 classes = checkpoint["classes"]
 
 # recreate model
@@ -25,7 +24,7 @@ model.load_state_dict(checkpoint["model_state"])
 model.to(device)
 model.eval()
 
-# preprocessing (SAME as training)
+# preprocessing
 transform = transforms.Compose([
     transforms.Resize((224, 224)),
     transforms.ToTensor(),
@@ -41,7 +40,3 @@ def predict(pil_image: Image.Image):
     logits = model(x)
     idx = logits.argmax(dim=1).item()
     return classes[idx]
-
-if __name__ == "__main__":
-    img = Image.open("Image.jpg")  # any face image
-    print(predict(img))