Update explain.py

explain.py

@@ -2,14 +2,14 @@ import matplotlib.pyplot as plt
 
 from visualization import generate_visualization
 
-def do_explain(transform, image, class_index=None, use_threshold=False):
+def do_explain(transform, image, class_index=None):
     fig, axs = plt.subplots(1, 2)
     axs[0].imshow(image)
     axs[0].axis("off")
 
     transformed_image = transform(image)
     viz = generate_visualization(
-        transformed_image, class_index=class_index, use_threshold=use_threshold
+        transformed_image, class_index=class_index
     )
 
     axs[1].imshow(viz)