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)
+    g = sns.heatmap(cm, cmap=cmap, cbar=True)
+    g.axes.get_yticklabels().set_color(text_color)
 
     # Add F1-score to the title
     plt.title(f"{title}\nF1 Score: {avg_f1:.3f}", color=text_color, fontsize=23)
@@ -96,8 +97,8 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path, theme='Dark'):
     plt.xticks(tick_marks, classes, color=text_color, fontsize=14)  # Adjust text color based on the mode
     plt.yticks(tick_marks, classes, color=text_color, fontsize=14)  # Adjust text color based on the mode
 
-    plt.ylabel('True label', color=text_color, fontsize=14)
-    plt.xlabel('Predicted label', color=text_color, fontsize=14)
+    plt.ylabel('True label', color=text_color, fontsize=20)
+    plt.xlabel('Predicted label', color=text_color, fontsize=20)
     plt.tight_layout()
 
     # Save the plot as an image