From bd3b18016d573959a05d7af3f6d6f2f32ebdb728 Mon Sep 17 00:00:00 2001
From: 彭震东 <zhendong.peng@qq.com>
Date: 星期二, 28 五月 2024 18:31:04 +0800
Subject: [PATCH] fix empty asr result (#1765)
---
funasr/auto/auto_model.py | 7 ++++---
1 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py
index 23d25c9..e8ca008 100644
--- a/funasr/auto/auto_model.py
+++ b/funasr/auto/auto_model.py
@@ -42,8 +42,9 @@
filelist = [".scp", ".txt", ".json", ".jsonl", ".text"]
chars = string.ascii_letters + string.digits
- if isinstance(data_in, str) and data_in.startswith("http://"): # url
- data_in = download_from_url(data_in)
+ if isinstance(data_in, str):
+ if data_in.startswith("http://") or data_in.startswith("https://"): # url
+ data_in = download_from_url(data_in)
if isinstance(data_in, str) and os.path.exists(
data_in
@@ -284,7 +285,7 @@
with torch.no_grad():
res = model.inference(**batch, **kwargs)
if isinstance(res, (list, tuple)):
- results = res[0]
+ results = res[0] if len(res) > 0 else [{"text": ""}]
meta_data = res[1] if len(res) > 1 else {}
time2 = time.perf_counter()
--
Gitblit v1.9.1