Update app.py

app.py

@@ -87,7 +87,8 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path, theme='Dark'):
     plt.figure(figsize=(10, 10))
 
     # Plot the confusion matrix with a colormap compatible for the mode
-    sns.heatmap(cm, cmap=cmap, cbar=True, linecolor=text_color) #, vmin=0, vmax=cm.max(), alpha=0.85)
+    fig, ax = plt.subplots()
+    sns.heatmap(cm, cmap=cmap, cbar=True, ax=ax, linecolor=text_color)
 
     cbar = ax.collections[0].colorbar
     cbar.ax.yaxis.set_tick_params(color=text_color)