Add BE model

This commit is contained in:
Eren Gölge 2023-09-04 13:57:03 +02:00
parent b4c82685a7
commit 562a9509f2
2 changed files with 11 additions and 8 deletions

View File

@ -720,7 +720,7 @@
"common-voice": { "common-voice": {
"glow-tts":{ "glow-tts":{
"description": "Belarusian GlowTTS model created by @alex73 (Github).", "description": "Belarusian GlowTTS model created by @alex73 (Github).",
"hf_url":"", "github_rls_url":"https://coqui.gateway.scarf.sh/v0.16.6/tts_models--be--common-voice--glow-tts.zip",
"default_vocoder": "vocoder_models/be/common-voice/hifigan", "default_vocoder": "vocoder_models/be/common-voice/hifigan",
"commit": "c0aabb85", "commit": "c0aabb85",
"license": "CC-BY-SA 4.0", "license": "CC-BY-SA 4.0",
@ -882,7 +882,7 @@
"be": { "be": {
"common-voice": { "common-voice": {
"hifigan": { "hifigan": {
"hf_url": "https://huggingface.co/coqui/hifigan-be", "github_rls_url": "https://coqui.gateway.scarf.sh/v0.16.6/vocoder_models--be--common-voice--hifigan.zip",
"description": "Belarusian HiFiGAN model created by @alex73 (Github).", "description": "Belarusian HiFiGAN model created by @alex73 (Github).",
"author": "@alex73", "author": "@alex73",
"license": "CC-BY-SA 4.0", "license": "CC-BY-SA 4.0",

View File

@ -468,13 +468,16 @@ class ModelManager(object):
print(f" > Error: Bad zip file - {file_url}") print(f" > Error: Bad zip file - {file_url}")
raise zipfile.BadZipFile # pylint: disable=raise-missing-from raise zipfile.BadZipFile # pylint: disable=raise-missing-from
# move the files to the outer path # move the files to the outer path
for file_path in z.namelist()[1:]: for file_path in z.namelist():
src_path = os.path.join(output_folder, file_path) src_path = os.path.join(output_folder, file_path)
dst_path = os.path.join(output_folder, os.path.basename(file_path)) if os.path.isfile(src_path):
if src_path != dst_path: dst_path = os.path.join(output_folder, os.path.basename(file_path))
copyfile(src_path, dst_path) if src_path != dst_path:
# remove the extracted folder copyfile(src_path, dst_path)
rmtree(os.path.join(output_folder, z.namelist()[0])) # remove redundant (hidden or not) folders
for file_path in z.namelist():
if os.path.isdir(os.path.join(output_folder, file_path)):
rmtree(os.path.join(output_folder, file_path))
@staticmethod @staticmethod
def _download_tar_file(file_url, output_folder, progress_bar): def _download_tar_file(file_url, output_folder, progress_bar):