Update utils/model_loader.py

utils/model_loader.py

@@ -34,8 +34,7 @@ class DenseNet121Classifier(nn.Module):
     
     def __init__(self, num_classes=2, pretrained=False):
         super(DenseNet121Classifier, self).__init__()
-        # self.densenet = models.densenet121(pretrained=pretrained)
-        models.densenet121(weights=models.DenseNet121_Weights.IMAGENET1K_V1)
+        self.densenet = models.densenet121(pretrained=pretrained)
         
         # Replace classifier layer
         num_ftrs = self.densenet.classifier.in_features