added normalization code

app.py

@@ -26,7 +26,11 @@ def inference(input_img, transparency = 0.5, target_layer_number = -1, num_top_c
         top: Top Classes and Confidence level of the prediction
         visualization: Grad Cam output
     """
-    transform = transforms.ToTensor()
+    # transform = transforms.ToTensor()
+    transform = transforms.Compose([
+                                    transforms.ToTensor(),
+                                    transforms.Normalize((0.1307,), (0.3081,))
+                                    ])
     org_img = input_img
     input_img = transform(input_img)
     # input_img = input_img