diff options
Diffstat (limited to 'swr2_asr/utils/decoder.py')
| -rw-r--r-- | swr2_asr/utils/decoder.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/swr2_asr/utils/decoder.py b/swr2_asr/utils/decoder.py index 1fd002a..38f218f 100644 --- a/swr2_asr/utils/decoder.py +++ b/swr2_asr/utils/decoder.py @@ -52,9 +52,11 @@ def get_beam_search_decoder( ) if not os.path.isdir(os.path.join(lang_model_path, f"mls_lm_{language}")): - url = f"https://dl.fbaipublicfiles.com/mls/mls_lm_{language}.tar.gz" - torch.hub.download_url_to_file(url, f"data/mls_lm_{language}.tar.gz") - _extract_tar("data/mls_lm_{language}.tar.gz", overwrite=True) + # check if zip file exists + if not os.path.isfile(f"data/mls_lm_{language}.tar.gz"): + url = f"https://dl.fbaipublicfiles.com/mls/mls_lm_{language}.tar.gz" + torch.hub.download_url_to_file(url, f"data/mls_lm_{language}.tar.gz") + _extract_tar(f"data/mls_lm_{language}.tar.gz", overwrite=True) tokens_path = os.path.join(lang_model_path, f"mls_lm_{language}", "tokens.txt") if not os.path.isfile(tokens_path): |