Update app.py

app.py

@@ -61,12 +61,11 @@ class_to_idx = {idx: cls for idx, cls in enumerate(classes)}
 train_transforms = torchvision.transforms.Compose(
     [
         torchvision.transforms.ToTensor(),
-        torchvision.transforms.Resize((TARGET_SIZE[0], TARGET_SIZE[1]), antialias=True),
+        torchvision.transforms.Resize((64, 64), antialias=True),
         CMYKToRGB(),
     ]
 )
 
-tf = Compose([torchvision.transforms.Resize((64, 64), antialias=True)])
 
 model = ViTIJEPA(64, 4, 3, 64, 8, 8, len(classes))
 model.load_state_dict(torch.load("vit_ijepa_ant_1.pt", map_location=torch.device('cpu')))