update app.py

app.py

@@ -3,7 +3,11 @@ import model_builder as mb
 from torchvision import transforms
 import torch 
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda" if torch.cuda.is_available() else "CPU"
+
+normalize = transforms.Normalize(
+        mean=[0.485, 0.456, 0.406], 
+        std=[0.229, 0.224, 0.225])
 
 manual_transform = transforms.Compose([
     transforms.ToPILImage(),