Upload folder using huggingface_hub

vgg16.py

@@ -52,7 +52,7 @@ VGG(
 class VGG_16(nn.Module):
     def __init__(self):
         super(VGG_16, self).__init__()
-        self.model = models.vgg16(weights='DEFAULT').features[:30]
+        self.model = models.vgg16(weights=models.VGG16_Weights).features[:30]
         
         for i, _ in enumerate(self.model):
             if i in [4, 9, 16, 23]:
@@ -60,7 +60,6 @@ class VGG_16(nn.Module):
                 
     def forward(self, x):
         features = []
-        
         for i, layer in enumerate(self.model):
             x = layer(x)
             if i in [0, 5, 10, 17, 24]:

vgg19.py

@@ -58,7 +58,7 @@ VGG(
 class VGG_19(nn.Module):
     def __init__(self):
         super(VGG_19, self).__init__()
-        self.model = models.vgg19(weights='DEFAULT').features[:30]
+        self.model = models.vgg19(weights=models.VGG19_Weights).features[:30]
         
         for i, _ in enumerate(self.model):
             if i in [4, 9, 18, 27]:
@@ -66,7 +66,6 @@ class VGG_19(nn.Module):
                 
     def forward(self, x):
         features = []
-        
         for i, layer in enumerate(self.model):
             x = layer(x)
             if i in [0, 5, 10, 19, 28]: