added mean and sd code for the image

app.py

@@ -26,12 +26,19 @@ 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((0.1307,), (0.3081,))
+                                    transforms.Normalize((mean_value,), (std_value,))
                                     ])
-    org_img = input_img
     input_img = transform(input_img)
     # input_img = input_img
     input_img = input_img.unsqueeze(0)