14 lines
687 B
Python
14 lines
687 B
Python
"""Explicit operator setup: download official models once, outside request handling."""
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
from modelscope.hub.snapshot_download import snapshot_download
|
|
root=Path(os.environ.get('SMART_EDIT_MODEL_DIR','.smart-edit-models')).resolve()
|
|
models={
|
|
'SMART_EDIT_ASR_MODEL':'iic/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch',
|
|
'SMART_EDIT_VAD_MODEL':'iic/speech_fsmn_vad_zh-cn-16k-common-pytorch',
|
|
'SMART_EDIT_PUNC_MODEL':'iic/punc_ct-transformer_zh-cn-common-vocab272727-pytorch',
|
|
}
|
|
result={key:snapshot_download(name,cache_dir=str(root)) for key,name in models.items()}
|
|
print(json.dumps(result,ensure_ascii=False,indent=2))
|