Update model.py

model.py

@@ -4,6 +4,7 @@ import torch.nn as nn
 import torchvision.models as models
 from PIL import Image
 from vgg_weights import VGG19_Weights
+wget https://github.com/cryu854/FastNeuralStyleTransfer/raw/master/models/vgg19-d01eb7cb.pth -P models/
 
 
 #Content Loss
@@ -88,9 +89,12 @@ def image_transform(image):
     model = torchvision.models.efficientnet_b2(weights=weights)
 
 #Defining a model
-weights=weights=VGG19_Weights.IMAGENET1K_V1
-cnn = models.vgg19(weights=weights).features.eval()
+# weights=weights=torchvision.models.VGG19_Weights.IMAGENET1K_V1
+# cnn = models.vgg19(weights=weights).features.eval()
+weights = models.vgg19(pretrained='imagenet')
+cnn = weights.features.eval()
 
+    
 #getting the input optimizer
 def get_input_optimizer(input_img):
     # this line to show that input is a parameter that requires a gradient