Update app.py

app.py

@@ -157,10 +157,10 @@ examples_dir = "sample"
 
 transformation_pipeline = transforms.Compose([
     transforms.ToPILImage(),
-    transforms.Grayscale(num_output_channels=1),
+    transforms.Grayscale(num_output_channels=3),
     transforms.CenterCrop((224, 224)),
     transforms.ToTensor(),
-    transforms.Normalize(mean=[0.485], std=[0.229])
+    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
 ])