Update model.py

model.py

@@ -65,6 +65,9 @@ cnn_normalization_std = torch.tensor([0.229, 0.224, 0.225])
 #     return image
 
 def image_transform(image):
+    if image is None:
+        return None
+
     if isinstance(image, str):
         # If image is a path to a file, open it using PIL
         with open(image, "rb") as f:
@@ -72,12 +75,14 @@ def image_transform(image):
     else:
         # If image is already a PIL image, just convert it to RGB mode
         image = image.convert('RGB')
+    
     # Apply the same transformations as before
     image = transform(image).unsqueeze(0)
     return image
 
 
 
+
 #Defining a model
 cnn = models.vgg19(pretrained=True).features.eval()