Read list of files from arguments

ls-milkyway · Jan 30, 2022 · d2c11a6 · d2c11a6
1 parent b090341
commit d2c11a6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/example/test_gpu_batch.py b/python/example/test_gpu_batch.py
@@ -14,7 +14,8 @@
 
 rec = BatchRecognizer()
 
-fnames = open("tedlium.list").readlines()
+# Read list of files from the file
+fnames = open(sys.argv[1]).readlines()
 fds = [open(x.strip(), "rb") for x in fnames]
 uids = [fname.strip().split('/')[-1][:-4] for fname in fnames]
 results = [""] * len(fnames)