Update app.py

app.py

@@ -71,7 +71,7 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path):
     plt.figure(figsize=(5, 5))
     
     # Plot the confusion matrix with a dark-mode compatible colormap
-    sns.heatmap(cm, annot=True, fmt=".2f", cmap="magma", cbar=False, annot_kws={"size": 12}, linewidths=0.5, linecolor='white')
+    sns.heatmap(cm, cmap="magma", cbar=False, linecolor='white')
     
     # Add F1-score to the title
     plt.title(f"{title}\n(F1 Score: {avg_f1:.3f})", color="white", fontsize=14)