zhifu gao
2024-04-24 861147c7308b91068ffa02724fdf74ee623a909e
funasr/bin/inference.py
@@ -21,11 +21,13 @@
    logging.basicConfig(level=log_level)
    if kwargs.get("debug", False):
        import pdb; pdb.set_trace()
        import pdb
        pdb.set_trace()
    model = AutoModel(**kwargs)
    res = model.generate(input=kwargs["input"])
    print(res)
if __name__ == '__main__':
if __name__ == "__main__":
    main_hydra()