enable running benchmark from anywhere

mobiusml · Jun 29, 2024 · 9f78b36 · 9f78b36
1 parent fcf0e82
commit 9f78b36
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/benchmark/wer_benchmark.py b/benchmark/wer_benchmark.py
@@ -1,6 +1,6 @@
 import argparse
 import json
-
+import os
 from datasets import load_dataset
 from evaluate import load
 from tqdm import tqdm
@@ -26,7 +26,9 @@
 
 # define the evaluation metric
 wer_metric = load("wer")
-normalizer = EnglishTextNormalizer(json.load(open("normalizer.json")))
+
+with open(os.path.join(os.path.dirname(__file__), "normalizer.json"), "r") as f:
+    normalizer = EnglishTextNormalizer(json.load(f))
 
 
 def inference(batch):