Update app.py

app.py

@@ -15,7 +15,7 @@ from sklearn.metrics import f1_score
 import seaborn as sns
 
 #################### BEAM PREDICTION #########################}
-def beam_prediction_task(data_percentage, task_complexity, theme):
+def beam_prediction_task(data_percentage, task_complexity, theme='Dark'):
     # Folder naming convention based on input_type, data_percentage, and task_complexity
     raw_folder = f"images/raw_{data_percentage/100:.1f}_{task_complexity}"
     embeddings_folder = f"images/embedding_{data_percentage/100:.1f}_{task_complexity}"
@@ -67,7 +67,7 @@ def compute_f1_score(cm):
     f1 = np.nan_to_num(f1)  # Replace NaN with 0
     return np.mean(f1)  # Return the mean F1-score across all classes
 
-def plot_confusion_matrix_beamPred(cm, classes, title, save_path, theme='Light'):
+def plot_confusion_matrix_beamPred(cm, classes, title, save_path, theme='Dark'):
     # Compute the average F1-score
     avg_f1 = compute_f1_score(cm)