Update model.py

model.py

@@ -53,13 +53,25 @@ class StyleLoss(nn.Module):
 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):
+#         # If image is a path to a file, open it using PIL
+#         image = Image.open(image).convert('RGB')
+#     else:
+#         # If image is a NumPy array, convert it to a PIL image
+#         image = Image.fromarray(image.astype('uint8'), 'RGB')
+#     # Apply the same transformations as before
+#     image = transform(image).unsqueeze(0)
+#     return image
+
 def image_transform(image):
     if isinstance(image, str):
         # If image is a path to a file, open it using PIL
-        image = Image.open(image).convert('RGB')
+        with open(image, "rb") as f:
+            image = Image.open(f).convert('RGB')
     else:
-        # If image is a NumPy array, convert it to a PIL image
-        image = Image.fromarray(image.astype('uint8'), 'RGB')
+        # 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