atsr

语帆

2024-03-01 fae6fd6d16e0fb060aa063790893f7555c421c4b

 funasr/bin/inference.py

@@ -1,7 +1,7 @@
import hydra
import logging
from omegaconf import DictConfig, OmegaConf, ListConfig

import pdb
from funasr.auto.auto_model import AutoModel


@@ -23,6 +23,7 @@
    if kwargs.get("debug", False):
        import pdb; pdb.set_trace()
    model = AutoModel(**kwargs)
    pdb.set_trace()
    res = model.generate(input=kwargs["input"])
    print(res)