funasr1.0

游雁

2023-12-21 b48251922fe347b3eb3768f633699fb8927ae71b

funasr1.0

 README_zh.md

@@ -81,11 +81,13 @@

### 非实时语音识别
```python
from funasr import infer
from funasr import AutoModel

p = infer(model="paraformer-zh", vad_model="fsmn-vad", punc_model="ct-punc", model_hub="ms")
model = AutoModel(model="paraformer-zh")
# for the long duration wav, you could add vad model
# model = AutoModel(model="paraformer-zh", vad_model="fsmn-vad")

res = p("asr_example_zh.wav", batch_size_token=5000)
res = model(input="asr_example_zh.wav", batch_size=5000)
print(res)
```
注：`model_hub`：表示模型仓库，`ms`为选择modelscope下载，`hf`为选择huggingface下载。

 funasr/__init__.py

@@ -29,3 +29,5 @@
    return results

import_submodules(__name__)

from funasr.bin.inference import AutoModel

 funasr/download/download_from_hub.py

@@ -12,6 +12,8 @@

def download_fr_ms(**kwargs):
    model_or_path = kwargs.get("model")
    if model_or_path in name_maps_ms:
        model_or_path = name_maps_ms[model_or_path]
    model_revision = kwargs.get("model_revision")
    if not os.path.exists(model_or_path):
        model_or_path = get_or_download_model_dir(model_or_path, model_revision, is_training=kwargs.get("is_training"))