Update app.py

app.py

@@ -68,7 +68,7 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path):
 
     # Set dark mode styling
     plt.style.use('dark_background')
-    plt.figure(figsize=(7, 7), facecolor='#2f2f2f')
+    plt.figure(figsize=(7, 7))
     
     # Plot the confusion matrix with a dark-mode compatible colormap
     sns.heatmap(cm, cmap="magma", cbar=True, linecolor='white')
@@ -77,15 +77,15 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path):
     plt.title(f"{title}\n(F1 Score: {avg_f1:.3f})", color="white", fontsize=14)
     
     tick_marks = np.arange(len(classes))
-    plt.xticks(tick_marks, classes, color="white", fontsize=10)  # White text for dark mode
-    plt.yticks(tick_marks, classes, color="white", fontsize=10)  # White text for dark mode
+    plt.xticks(tick_marks, classes, color="white", fontsize=12)  # White text for dark mode
+    plt.yticks(tick_marks, classes, color="white", fontsize=12)  # White text for dark mode
 
     plt.ylabel('True label', color="white", fontsize=12)
     plt.xlabel('Predicted label', color="white", fontsize=12)
     plt.tight_layout()
     
     # Save the plot as an image
-    plt.savefig(save_path, facecolor='#2f2f2f', transparent=True)  # Use transparent to blend with the dark mode website
+    plt.savefig(save_path, transparent=True)  # Use transparent to blend with the dark mode website
     plt.close()
     
     # Return the saved image