WwYc commited on
Commit
18c84c1
·
verified ·
1 Parent(s): 44c214e

Update explain.py

Browse files
Files changed (1) hide show
  1. explain.py +1 -1
explain.py CHANGED
@@ -14,5 +14,5 @@ def do_explain(transform, image, class_index=None):
14
  predict = print_top_classes(transformed_image)
15
  axs[1].imshow(viz)
16
  axs[1].axis("off")
17
- return fig predict
18
 
 
14
  predict = print_top_classes(transformed_image)
15
  axs[1].imshow(viz)
16
  axs[1].axis("off")
17
+ return fig, predict
18