Update app.py

app.py

@@ -38,6 +38,12 @@ model.load_state_dict(state)
 device = torch.device('cuda' if USE_CUDA else 'cpu')
 model = model.to(device)
 
+transform = torchvision.transforms.Compose([
+    torchvision.transforms.Resize((224, 224)),
+    torchvision.transforms.ToTensor(),
+    torchvision.transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.225, 0.225, 0.225])
+])
+
 classes = ["chmury", 'inne', "dym"]
 
 def predict(image, model=model, classes=classes, device=device, transform=transform):