Fix weights path

source/predict_sample.py

@@ -45,7 +45,7 @@ def main_classification(image):
     cnn = CNN().to(torch.device("cpu"))
     cnn.eval()
 
-    cnn.load_state_dict(torch.load(f='weights/CNN-B8-LR-0.01-E30.pt', map_location=torch.device("cpu")))
+    cnn.load_state_dict(torch.load(f='source/weights/CNN-B8-LR-0.01-E30.pt', map_location=torch.device("cpu")))
 
     prediction_outcome = classify_eye(image, cnn)