15 lines
315 B
Python
15 lines
315 B
Python
|
'''
|
||
|
|
Pre-fetches the required model from github.
|
||
|
|
'''
|
||
|
|
|
||
|
|
import os
|
||
|
|
from spleeter.model.provider import get_default_model_provider
|
||
|
|
|
||
|
|
|
||
|
|
PREFETCH_MODELS = ['2stems']
|
||
|
|
|
||
|
|
for model_name in PREFETCH_MODELS:
|
||
|
|
get_default_model_provider().download(
|
||
|
|
model_name,
|
||
|
|
os.path.join(os.getenv('MODEL_PATH'), model_name))
|