Update app.py

app.py

@@ -48,7 +48,7 @@ def inference(input_img, transparancy = 0.5, target_layer_number = -1):
     prediction= torch.max(outputs, 1)
     target_layers = [model.layer2[target_layer_number]]
     cam = GradCAM(model=model, target_layers=target_layers)
-    grayscale_cam = cam(input_tensor= input_img,target=None)
+    grayscale_cam = cam(input_tensor= input_img)
     grayscale_cam = grayscale_cam[0, :]
     visualization = show_cam_on_image(org_img/255,grayscale_cam,use_rgb=True,
                                      image_weight=transparancy)