Update app.py

app.py

@@ -71,7 +71,8 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path):
     plt.figure(figsize=(10, 10))
     
     # Plot the confusion matrix with a dark-mode compatible colormap
-    sns.heatmap(cm, cmap="magma", cbar=True, linecolor='white', vmin=0, vmax=cm.max(), alpha=0.85)
+    #sns.heatmap(cm, cmap="magma", cbar=True, linecolor='white', vmin=0, vmax=cm.max(), alpha=0.85)
+    sns.heatmap(cm, cmap="cividis", cbar=True, linecolor='white', vmin=0, vmax=cm.max(), alpha=0.85)
     
     # Add F1-score to the title
     plt.title(f"{title}\n(F1 Score: {avg_f1:.3f})", color="white", fontsize=14)