Update model.py

no gpu so updated the device agnoistic code

model.py

@@ -46,8 +46,8 @@ class StyleLoss(nn.Module):
 
 #Normalization
 
-cnn_normalization_mean = torch.tensor([0.485, 0.456, 0.406]).to(device)
-cnn_normalization_std = torch.tensor([0.229, 0.224, 0.225]).to(device)
+cnn_normalization_mean = torch.tensor([0.485, 0.456, 0.406])
+cnn_normalization_std = torch.tensor([0.229, 0.224, 0.225])
 #image transformation
 def image_transform(image):
     if isinstance(image, str):
@@ -58,12 +58,12 @@ def image_transform(image):
         image = Image.fromarray(image.astype('uint8'), 'RGB')
     # Apply the same transformations as before
     image = transform(image).unsqueeze(0)
-    return image.to(device)
+    return image
 
 
 
 #Defining a model
-cnn = models.vgg19(pretrained=True).features.to(device).eval()
+cnn = models.vgg19(pretrained=True).features.eval()
 
 #getting the input optimizer
 def get_input_optimizer(input_img):
@@ -82,7 +82,7 @@ def get_style_model_and_losses(cnn, normalization_mean, normalization_std,
                                content_layers=content_layers_default,
                                style_layers=style_layers_default):
     # normalization module
-    normalization = Normalization(normalization_mean, normalization_std).to(device)
+    normalization = Normalization(normalization_mean, normalization_std)
 
     # just in order to have an iterable access to or list of content/style
     # losses