removed mean and std calculation due to poor prediction

app.py

@@ -26,19 +26,12 @@ 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
     """
-
-    org_img = input_img
-    # Convert the image to a numerical array
-    image_array = np.array(input_img, dtype=np.float32)
-
-    # Calculate the mean and standard deviation
-    mean_value = np.mean(image_array)
-    std_value = np.std(image_array)
     # transform = transforms.ToTensor()
     transform = transforms.Compose([
                                     transforms.ToTensor(),
-                                    transforms.Normalize((mean_value,), (std_value,))
+                                    transforms.Normalize((0.1307,), (0.3081,))
                                     ])
+    org_img = input_img
     input_img = transform(input_img)
     # input_img = input_img
     input_img = input_img.unsqueeze(0)