Manu101 commited on
Commit
190826e
·
verified ·
1 Parent(s): df0fc51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def inference(input_image, transparency=0.5, target_layer_number=-1):
55
  _, prediction = torch.max(outputs, 1)
56
  target_layers = [model.layer2[target_layer_number]]
57
  cam = GradCAM(model= model, target_layers = target_layers)
58
- grayscale_cam = cam(input_tensor=input_image, target=None)
59
  grayscale_cam = grayscale_cam[0, :]
60
  visualization = show_cam_on_image(
61
  org_img/255,
 
55
  _, prediction = torch.max(outputs, 1)
56
  target_layers = [model.layer2[target_layer_number]]
57
  cam = GradCAM(model= model, target_layers = target_layers)
58
+ grayscale_cam = cam(input_tensor=input_image, targets=None)
59
  grayscale_cam = grayscale_cam[0, :]
60
  visualization = show_cam_on_image(
61
  org_img/255,