Update explain.py

explain.py

@@ -14,5 +14,5 @@ def do_explain(transform, image, class_index=None):
     predict = print_top_classes(transformed_image)
     axs[1].imshow(viz)
     axs[1].axis("off")
-    return fig
+    return fig， predict