done ig

app.py

@@ -46,7 +46,7 @@ def inference(input_img, transparency = 0.5, target_layer_number = -1):
     transform = transforms.ToTensor()
 
     input_img = transform(input_img)
-    # input_img = cifar10_normalization()(input_img)
+    input_img = cifar10_normalization()(input_img)
     input_img = input_img.unsqueeze(0)
     outputs = model(input_img)
     softmax = torch.nn.Softmax(dim=0)
@@ -61,7 +61,7 @@ def inference(input_img, transparency = 0.5, target_layer_number = -1):
     grayscale_cam = grayscale_cam[0, :]
 
     img = input_img.squeeze(0)
-    img = inv_normalize(img)
+    # img = inv_normalize(img)
     print(transparency)
 
     visualization = show_cam_on_image(org_img/255, grayscale_cam, use_rgb=True, image_weight=transparency)