Update app.py

app.py

@@ -49,7 +49,12 @@ DesignModernityModel.eval() # set state of the model to inference
 
 LABELS = ['2000-2004', '2006-2008', '2009-2011', '2012-2015', '2016-2018']
 
-carTransforms = transforms.Compose([transforms.Resize(224)])
+MEAN = [0.485, 0.456, 0.406]
+STD = [0.229, 0.224, 0.225]
+
+carTransforms = transforms.Compose([transforms.Resize(224),
+                                    transforms.ToTensor(),
+                                    transforms.Normalize(mean=MEAN, std=STD)])
 
 def classifyCar(im):
   im = Image.fromarray(im.astype('uint8'), 'RGB')