Spaces:
Sleeping
Sleeping
Update explain.py
Browse files- 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
|
18 |
|
|
|
14 |
predict = print_top_classes(transformed_image)
|
15 |
axs[1].imshow(viz)
|
16 |
axs[1].axis("off")
|
17 |
+
return fig, predict
|
18 |
|