zhifu gao
2024-01-18 b28f3c9da94ae72a3a0b7bb5982b587be7cf4cd6
examples/industrial_data_pretraining/emotion2vec/demo.py
@@ -5,7 +5,8 @@
from funasr import AutoModel
model = AutoModel(model="../modelscope_models/emotion2vec_base")
model = AutoModel(model="damo/emotion2vec_base", model_revision="v2.0.1")
res = model(input="../modelscope_models/emotion2vec_base/example/test.wav")
wav_file = f"{model.model_path}/example/test.wav"
res = model.generate(wav_file, output_dir="./outputs", granularity="utterance")
print(res)