Fix install_models.py (2)

This commit is contained in:
Piero Toffanin 2022-04-01 10:06:03 -04:00 committed by GitHub
parent 8cc03572e5
commit a2bea5e300
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -7,7 +7,7 @@ if __name__ == "__main__":
parser.add_argument("--load_only_lang_codes", type=str, default="")
args = parser.parse_args()
lang_codes = args.load_only_lang_codes.split(",")
if len(lang_codes) == 0:
if len(lang_codes) == 0 or lang_codes[0] == '':
lang_codes = None
check_and_install_models(force=True, load_only_lang_codes=lang_codes)
check_and_install_transliteration(force=True)