Update model.py

added weights to the vgg model

model.py

@@ -81,10 +81,14 @@ def image_transform(image):
     return image
 
 
-
+    # Create EffNetB2 pretrained weights, transforms and model
+    weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
+    transforms = weights.transforms()
+    model = torchvision.models.efficientnet_b2(weights=weights)
 
 #Defining a model
-cnn = models.vgg19(pretrained=True).features.eval()
+weights=weights=VGG19_Weights.IMAGENET1K_V1
+cnn = models.vgg19(weights=weights).features.eval()
 
 #getting the input optimizer
 def get_input_optimizer(input_img):