refines

app.py

@@ -305,6 +305,24 @@ def get_wer_metrics(dataset):
         nb_oracle_row["test_ls_other"] = 2.7 / 100
         cp_oracle_row["test_ls_other"] = 1.6 / 100
     
+    # Calculate overall averages for the three additional methods
+    llama_values = []
+    nb_oracle_values = []
+    cp_oracle_values = []
+
+    for source in all_sources:
+        if pd.notna(llama_lora_row[source]):
+            llama_values.append(llama_lora_row[source])
+        if pd.notna(nb_oracle_row[source]):
+            nb_oracle_values.append(nb_oracle_row[source])
+        if pd.notna(cp_oracle_row[source]):
+            cp_oracle_values.append(cp_oracle_row[source])
+
+    # Calculate and set the overall values
+    llama_lora_row["OVERALL"] = np.mean(llama_values) if llama_values else np.nan
+    nb_oracle_row["OVERALL"] = np.mean(nb_oracle_values) if nb_oracle_values else np.nan
+    cp_oracle_row["OVERALL"] = np.mean(cp_oracle_values) if cp_oracle_values else np.nan
+    
     # Add rows in the desired order
     rows.append(no_lm_row)
     rows.append(lm_ranking_row)