Spaces:
Runtime error
Runtime error
path
Browse files
pages/Model_Evaluation.py
CHANGED
|
@@ -105,12 +105,12 @@ def load_test_data(csv_path):
|
|
| 105 |
def load_model():
|
| 106 |
model = models.densenet121(pretrained=False)
|
| 107 |
model.classifier = nn.Linear(model.classifier.in_features, len(class_names))
|
| 108 |
-
model.load_state_dict(torch.load(r"D
|
| 109 |
model.eval()
|
| 110 |
return model
|
| 111 |
|
| 112 |
# ---- Main UI Buttons ----
|
| 113 |
-
csv_path = r"D
|
| 114 |
model = load_model()
|
| 115 |
test_loader = load_test_data(csv_path)
|
| 116 |
|
|
|
|
| 105 |
def load_model():
|
| 106 |
model = models.densenet121(pretrained=False)
|
| 107 |
model.classifier = nn.Linear(model.classifier.in_features, len(class_names))
|
| 108 |
+
model.load_state_dict(torch.load(r"D:\DR_Classification\training\Pretrained_Densenet-121.pth", map_location=torch.device('cpu')))
|
| 109 |
model.eval()
|
| 110 |
return model
|
| 111 |
|
| 112 |
# ---- Main UI Buttons ----
|
| 113 |
+
csv_path = r"D:\DR_Classification\splits\test_labels.csv"
|
| 114 |
model = load_model()
|
| 115 |
test_loader = load_test_data(csv_path)
|
| 116 |
|