def test_model():
d = ModelDownloader()
tasks = ["asr", "tts"]
for task in tasks:
for model_name in d.query(task=task):
if d.query("valid", name=model_name)[0] == "false":
continue
print(f"#### Test {model_name} ####")
if task == "asr":
_asr(model_name)
elif task == "tts":
_tts(model_name)
else:
raise NotImplementedError(f"task={task}")
def test_model():
d = ModelDownloader()
tasks = ["asr", "tts"]
for task in tasks:
for corpus in list(set(d.query("corpus", task=task))):
for model_name in d.query(task=task, corpus=corpus):
if d.query("valid", name=model_name)[0] == "false":
continue
print(f"#### Test {model_name} ####")
if task == "asr":
_asr(model_name)
elif task == "tts":
_tts(model_name)
else:
raise NotImplementedError(f"task={task}")
# NOTE(kan-bayashi): remove and recreate cache dir to reduce the disk usage.
shutil.rmtree("downloads")
os.makedirs("downloads")
def create_Readme_file(repo_name, model_name):
# Fill in the blanks in the template Readme eg. add task tags, model name etc.
d = ModelDownloader()
corpus_name = d.query("corpus", name=model_name)[0]
task_name = d.query("task", name=model_name)[0]
url_name = d.query("url", name=model_name)[0].split("files/")[0]
user_name = model_name.split("/")[0]
lang_name = d.query("lang", name=model_name)[0].replace("jp", "ja")
template_Readme = open("TEMPLATE_Readme.md")
new_Readme = open(repo_name + "/README.md", "w")
lines_arr = [line for line in template_Readme]
line_final_arr = []
for line in lines_arr:
if "<add_more_tags>" in line:
if task_name == "asr":
line = line.replace("<add_more_tags>",
"automatic-speech-recognition")
elif task_name == "tts":
line = line.replace("<add_more_tags>", "text-to-speech")
elif task_name == "enh":
line = line.replace("<add_more_tags>",
"speech-enhancement\n- audio-to-audio")
if "<add_lang>" in line:
if lang_name == "multilingual":
line = line.replace("<add_lang>",
"en\n- zh\n- ja\n- multilingual")
else:
line = line.replace("<add_lang>", lang_name)
line = line.replace("<add_model_name>", model_name)
line = line.replace("<add_url>", url_name)
line = line.replace("<add_name>", user_name)
line = line.replace("<add_corpus>", corpus_name)
line = line.replace("<add_task_name>", task_name.upper())
line = line.replace("<add_recipe_task_name>", task_name.lower() + "1")
if "<add_tts_reference>" in line:
if task_name == "tts":
line = line.replace("<add_tts_reference>", tts_reference)
else:
line = line.replace("<add_tts_reference>", "")
new_Readme.write(line)
def test_get_model_names_non_matching():
d = ModelDownloader()
assert d.query("name", task="dummy") == []
def test_get_model_names_and_urls():
d = ModelDownloader()
d.query(["name", "url"], task="asr")
def test_download_and_unpack_names_with_condition():
d = ModelDownloader()
d.query("name", task="asr")