From 583f918e0ec752518a0263b09ce9b9e55f047fa4 Mon Sep 17 00:00:00 2001
From: 语帆 <yf352572@alibaba-inc.com>
Date: 星期五, 01 三月 2024 11:18:29 +0800
Subject: [PATCH] atsr

---
 funasr/models/lcbnet/model.py |    5 ++++-
 1 files changed, 4 insertions(+), 1 deletions(-)

diff --git a/funasr/models/lcbnet/model.py b/funasr/models/lcbnet/model.py
index 09e6dd1..e83f8d7 100644
--- a/funasr/models/lcbnet/model.py
+++ b/funasr/models/lcbnet/model.py
@@ -428,7 +428,10 @@
             meta_data["load_data"] = f"{time2 - time1:0.3f}"
             pdb.set_trace()
             audio_sample_list = sample_list[0]
-            ocr_sample_list = sample_list[1]
+            if len(sample_list) >1:
+                ocr_sample_list = sample_list[1]
+            else:
+                ocr_sample_list = [294, 0]
             speech, speech_lengths = extract_fbank(audio_sample_list, data_type=kwargs.get("data_type", "sound"),
                                                    frontend=frontend)
             time3 = time.perf_counter()

--
Gitblit v1.9.1