Update model.py

model.py

@@ -5,6 +5,16 @@ import torch.nn as nn
 import torchvision.models as models
 from PIL import Image
 from torchvision import transforms
+import time
+import torch.nn.functional as F
+import torch.optim as optim
+import matplotlib.pyplot as plt
+import torchvision.transforms as transforms
+import copy
+import torchvision.models as models
+import torchvision.transforms.functional as TF
+from PIL import Image
+import numpy as np
 
 # from vgg_weights import VGG19_Weights
 
@@ -70,8 +80,7 @@ def image_transform(image):
 
 
         
-#Defining the Model
-cnn = models.vgg19(pretrained=True).features.eval()
+
 
 #Normalization