Spaces:
Running
Running
File size: 1,036 Bytes
e976963 21bd41f e976963 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
from pathlib import Path
import requests
RVC_other_DOWNLOAD_LINK = 'https://huggingface.co/Politrees/RVC_resources/resolve/main/predictors/'
RVC_hubert_DOWNLOAD_LINK = 'https://huggingface.co/Politrees/RVC_resources/resolve/main/embedders/'
BASE_DIR = Path(__file__).resolve().parent.parent
rvc_models_dir = BASE_DIR / 'rvc_models'
def dl_model(link, model_name, dir_name):
with requests.get(f'{link}{model_name}') as r:
r.raise_for_status()
with open(dir_name / model_name, 'wb') as f:
for chunk in r.iter_content(chunk_size=8192):
f.write(chunk)
if __name__ == '__main__':
rvc_other_names = ['rmvpe.pt', 'fcpe.pt']
for model in rvc_other_names:
print(f'Downloading {model}...')
dl_model(RVC_other_DOWNLOAD_LINK, model, rvc_models_dir)
rvc_hubert_names = ['hubert_base.pt']
for model in rvc_hubert_names:
print(f'Downloading {model}...')
dl_model(RVC_hubert_DOWNLOAD_LINK, model, rvc_models_dir)
print('All models downloaded!')
|