Hugging Face's logo

Spaces:
madhurjindal
/
image_classification_cifar10_gradcam
Sleeping

App Files Community
madhurjindal commited on
Commit
56082ef
·
1 Parent(s): a27f812

Update Utilities/visualize.py

Browse files
Files changed (1) hide show
  1. Utilities/visualize.py +1 -0
Utilities/visualize.py CHANGED
@@ -30,4 +30,5 @@ def plot_incorrect_preds(incorrect, classes, num_imgs):
30
  plt.xticks([])
31
  plt.yticks([])
32
  plt.tight_layout()
 
33
  return fig
 
30
  plt.xticks([])
31
  plt.yticks([])
32
  plt.tight_layout()
33
+ plt.close()
34
  return fig