| | |
|---|
| | | |
|---|
| | | from funasr import AutoModel |
|---|
| | | |
|---|
| | | model = AutoModel(model="damo/emotion2vec_base", model_revision="v2.0.1") |
|---|
| | | # model="iic/emotion2vec_base" |
|---|
| | | # model="iic/emotion2vec_base_finetuned" |
|---|
| | | # model="iic/emotion2vec_plus_seed" |
|---|
| | | # model="iic/emotion2vec_plus_base" |
|---|
| | | model = "iic/emotion2vec_plus_large" |
|---|
| | | |
|---|
| | | model = AutoModel( |
|---|
| | | model=model, |
|---|
| | | # vad_model="iic/speech_fsmn_vad_zh-cn-16k-common-pytorch", |
|---|
| | | # vad_model_revision="master", |
|---|
| | | # vad_kwargs={"max_single_segment_time": 2000}, |
|---|
| | | hub="ms", # "ms" or "modelscope" for Mainland China users; "hf" or "huggingface" for Other overseas users |
|---|
| | | ) |
|---|
| | | |
|---|
| | | wav_file = f"{model.model_path}/example/test.wav" |
|---|
| | | res = model.generate(wav_file, output_dir="./outputs", granularity="utterance") |
|---|
| | | print(res) |
|---|
| | | |
|---|
| | | res = model.generate( |
|---|
| | | wav_file, output_dir="./outputs", granularity="utterance", extract_embedding=False |
|---|
| | | ) |
|---|
| | | print(res) |
|---|
