WwYc commited on
Commit
c32bf50
·
verified ·
1 Parent(s): 0b54afc

Update explain.py

Browse files
Files changed (1) hide show
  1. explain.py +1 -1
explain.py CHANGED
@@ -9,7 +9,7 @@ def do_explain(transform, image, class_index=None, use_threshold=False):
9
 
10
  transformed_image = transform(image)
11
  viz = generate_visualization(
12
- transformed_image, class_index=class_index, method="gradcam", use_threshold=False
13
  )
14
 
15
  axs[1].imshow(viz)
 
9
 
10
  transformed_image = transform(image)
11
  viz = generate_visualization(
12
+ transformed_image, class_index=class_index, use_threshold=use_threshold
13
  )
14
 
15
  axs[1].imshow(viz)