26 lines
545 B
Python
26 lines
545 B
Python
import torch
|
|
|
|
from model import FunASRNano
|
|
|
|
|
|
def main():
|
|
model_dir = "FunAudioLLM/Fun-ASR-Nano-2512"
|
|
device = (
|
|
"cuda:0"
|
|
if torch.cuda.is_available()
|
|
else "mps"
|
|
if torch.backends.mps.is_available()
|
|
else "cpu"
|
|
)
|
|
m, kwargs = FunASRNano.from_pretrained(model=model_dir, device=device)
|
|
m.eval()
|
|
|
|
wav_path = f"{kwargs['model_path']}/example/zh.mp3"
|
|
res = m.inference(data_in=[wav_path], **kwargs)
|
|
text = res[0][0]["text"]
|
|
print(text)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|