From 93a02feda1848531a48a1794a8de12a962c84a8e Mon Sep 17 00:00:00 2001
From: Xingchen Song(宋星辰) <xingchensong1996@163.com>
Date: 星期五, 07 六月 2024 23:10:13 +0800
Subject: [PATCH] [fix] fix empty asr result (#1794)
---
funasr/__init__.py | 36 +++++++++++++++++++++++++++++++++++-
1 files changed, 35 insertions(+), 1 deletions(-)
diff --git a/funasr/__init__.py b/funasr/__init__.py
index d0b7aa5..c7f7c21 100644
--- a/funasr/__init__.py
+++ b/funasr/__init__.py
@@ -1,10 +1,44 @@
"""Initialize funasr package."""
import os
+import pkgutil
+import importlib
dirname = os.path.dirname(__file__)
version_file = os.path.join(dirname, "version.txt")
with open(version_file, "r") as f:
__version__ = f.read().strip()
-from funasr.bin.inference_cli import infer
\ No newline at end of file
+
+import importlib
+import pkgutil
+
+
+def import_submodules(package, recursive=True):
+ if isinstance(package, str):
+ try:
+ package = importlib.import_module(package)
+ except Exception as e:
+ # 濡傛灉鎯宠鐪嬪埌瀵煎叆閿欒鐨勫叿浣撲俊鎭紝鍙互鍙栨秷娉ㄩ噴涓嬮潰鐨勮
+ # print(f"Failed to import {package}: {e}")
+ pass
+ results = {}
+ if not isinstance(package, str):
+ for loader, name, is_pkg in pkgutil.walk_packages(package.__path__, package.__name__ + "."):
+ try:
+ results[name] = importlib.import_module(name)
+ except Exception as e:
+ # 濡傛灉鎯宠鐪嬪埌瀵煎叆閿欒鐨勫叿浣撲俊鎭紝鍙互鍙栨秷娉ㄩ噴涓嬮潰鐨勮
+ # print(f"Failed to import {name}: {e}")
+ pass
+ if recursive and is_pkg:
+ results.update(import_submodules(name))
+ return results
+
+
+import_submodules(__name__)
+
+from funasr.auto.auto_model import AutoModel
+from funasr.auto.auto_frontend import AutoFrontend
+
+os.environ["HYDRA_FULL_ERROR"] = "1"
--
Gitblit v1.9.1