Spaces:
Running
Running
Commit
·
5e78dfc
1
Parent(s):
82994f7
SW-202 add more descriptive comments and fix spelling
Browse files
main.py
CHANGED
@@ -116,11 +116,11 @@ model_to_multilingual_wer = dict(
|
|
116 |
# Add English WER and Multilingual WER to performance_df
|
117 |
benchmark_df["english_wer"] = benchmark_df["model"].map(model_to_english_wer)
|
118 |
benchmark_df["multilingual_wer"] = benchmark_df["model"].map(model_to_multilingual_wer)
|
119 |
-
benchmark_df.fillna({"multilingual_wer": "N/A"}, inplace=True)
|
120 |
|
121 |
-
#
|
122 |
english_only_mask = benchmark_df["model"].str.contains(r"\.en$|distil-whisper", case=False, na=False)
|
123 |
-
benchmark_df.loc[english_only_mask, "multilingual_wer"] = "English
|
124 |
|
125 |
benchmark_df["multilingual_wer"] = benchmark_df["multilingual_wer"].astype(str)
|
126 |
|
|
|
116 |
# Add English WER and Multilingual WER to performance_df
|
117 |
benchmark_df["english_wer"] = benchmark_df["model"].map(model_to_english_wer)
|
118 |
benchmark_df["multilingual_wer"] = benchmark_df["model"].map(model_to_multilingual_wer)
|
119 |
+
benchmark_df.fillna({"multilingual_wer": "N/A"}, inplace=True) # Mark all untested models as N/A
|
120 |
|
121 |
+
# Mark English-only models
|
122 |
english_only_mask = benchmark_df["model"].str.contains(r"\.en$|distil-whisper", case=False, na=False)
|
123 |
+
benchmark_df.loc[english_only_mask, "multilingual_wer"] = "English-only model"
|
124 |
|
125 |
benchmark_df["multilingual_wer"] = benchmark_df["multilingual_wer"].astype(str)
|
126 |
|