save

pages/Model_Evaluation.py

@@ -105,12 +105,12 @@ def load_test_data(csv_path):
 def load_model():
     model = models.densenet121(pretrained=False)
     model.classifier = nn.Linear(model.classifier.in_features, len(class_names))
-    model.load_state_dict(torch.load(r"D:\DR_Classification\training\Pretrained_Densenet-121.pth", map_location=torch.device('cpu')))
+    model.load_state_dict(torch.load(r"training\Pretrained_Densenet-121.pth", map_location=torch.device('cpu')))
     model.eval()
     return model
 
 # ---- Main UI Buttons ----
-csv_path = r"D:\DR_Classification\splits\test_labels.csv"
+csv_path = r"splits\test_labels.csv"
 model = load_model()
 test_loader = load_test_data(csv_path)