Spaces:
Runtime error
Runtime error
| import torch | |
| import os | |
| from . import utils | |
| cwd = os.getcwd() | |
| DOWNLOAD_CKPT_URLS = { | |
| 'EN': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/EN/checkpoint.pth', | |
| 'EN_V2': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/EN_V2/checkpoint.pth', | |
| 'FR': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/FR/checkpoint.pth', | |
| 'JP': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/JP/checkpoint.pth', | |
| 'ES': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/ES/checkpoint.pth', | |
| 'ZH': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/ZH/checkpoint.pth', | |
| 'KR': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/KR/checkpoint.pth', | |
| } | |
| DOWNLOAD_CONFIG_URLS = { | |
| 'EN': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/EN/config.json', | |
| 'EN_V2': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/EN_V2/config.json', | |
| 'FR': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/FR/config.json', | |
| 'JP': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/JP/config.json', | |
| 'ES': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/ES/config.json', | |
| 'ZH': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/ZH/config.json', | |
| 'KR': 'https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/basespeakers/KR/config.json', | |
| } | |
| def load_or_download_config(locale): | |
| language = locale.split('-')[0].upper() | |
| assert language in DOWNLOAD_CONFIG_URLS | |
| #config_path = os.path.expanduser(f'~/.local/share/openvoice/basespeakers/{language}/config.json') | |
| config_path = os.path.join(cwd, f"openvoice/basespeakers/{language}/config.json") | |
| try: | |
| return utils.get_hparams_from_file(config_path) | |
| except: | |
| # download | |
| os.makedirs(os.path.dirname(config_path), exist_ok=True) | |
| #os.system(f'wget {DOWNLOAD_CONFIG_URLS[language]} -O {config_path}') | |
| print(f"URL = {DOWNLOAD_CONFIG_URLS[language]}") | |
| os.system(f"curl -o {config_path} {DOWNLOAD_CONFIG_URLS[language]}") | |
| return utils.get_hparams_from_file(config_path) | |
| def load_or_download_model(locale, device): | |
| language = locale.split('-')[0].upper() | |
| assert language in DOWNLOAD_CKPT_URLS | |
| #ckpt_path = os.path.expanduser(f'~/.local/share/openvoice/basespeakers/{language}/checkpoint.pth') | |
| ckpt_path = os.path.join(cwd, f"openvoice/basespeakers/{language}/checkpoint.pth") | |
| try: | |
| return torch.load(ckpt_path, map_location=device) | |
| except: | |
| # download | |
| os.makedirs(os.path.dirname(ckpt_path), exist_ok=True) | |
| #os.system(f'wget {DOWNLOAD_CKPT_URLS[language]} -O {ckpt_path}') | |
| os.system(f"curl -o {ckpt_path} {DOWNLOAD_CKPT_URLS[language]}") | |
| return torch.load(ckpt_path, map_location=device) | |