From 2892a70cd847b020be44460e90f52091ddc1dc95 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BF=97=E6=B5=A9?= <neo.dzh@alibaba-inc.com>
Date: Tue, 24 Sep 2024 23:10:49 +0800
Subject: [PATCH] add extract_token binary

---
 funasr/bin/extract_token.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/funasr/bin/extract_token.py b/funasr/bin/extract_token.py
index 1bc6243b2..2e04d3a6f 100644
--- a/funasr/bin/extract_token.py
+++ b/funasr/bin/extract_token.py
@@ -42,7 +42,7 @@ def main_hydra(cfg: DictConfig):
     torch.cuda.set_device(int(os.environ["LOCAL_RANK"]))
     model = AutoModel(**kwargs)
     res = model.generate(input=kwargs["input"])
-    print(res)
+    logging.info(f"{res}")
 
 
 if __name__ == "__main__":