Cloudy1225 commited on
Commit
f6e244e
·
1 Parent(s): 38f046a

Add `plt.close()` to save memory.

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -32,6 +32,7 @@ def classify(num: int):
32
  samples_df['Predict'] = y_pred
33
  bar = plot_bar(y_pred.value_counts())
34
  cm = plot_confusion_matrix(y_pred, y)
 
35
  return samples_df, bar, cm
36
 
37
 
 
32
  samples_df['Predict'] = y_pred
33
  bar = plot_bar(y_pred.value_counts())
34
  cm = plot_confusion_matrix(y_pred, y)
35
+ plt.close()
36
  return samples_df, bar, cm
37
 
38