Update app.py

app.py

@@ -72,7 +72,7 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path, theme=None):
     avg_f1 = compute_f1_score(cm)
 
     # Choose the color scheme based on the user's mode
-    if theme == 'dark':
+    if theme == 'Dark':
         plt.style.use('dark_background')  # Use dark mode styling
         text_color = 'white'
         cmap = 'cividis'  # Dark-mode-friendly colormap