Spaces:

WwYc
/

explain-ViT

Sleeping

WwYc commited on Mar 4, 2024

Commit

0b54afc

verified ·

1 Parent(s): a041d6e

Update explain.py

Files changed (1) hide show

explain.py CHANGED Viewed

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

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