Update preprocess.py

preprocess.py

@@ -7,9 +7,9 @@ from PIL import Image
 def preprocess_image(example, image_size):
     image = example['image'].convert('RGB')  # Directly use the PIL image
     transform = transforms.Compose([
-        transforms.Resize((image_size, image_size)),
+        #transforms.Resize((image_size, image_size)),
         transforms.ToTensor(),
-        transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
+        #transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
     ])
     image = transform(image)
     return {'image': image, 'label': example['label']}