From 1e5ef6ed9a6f64ecca7b9ef9481519b271f793a3 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 23 十二月 2024 19:06:50 +0800
Subject: [PATCH] bug fix
---
funasr/utils/load_utils.py | 13 +++++++++++--
1 files changed, 11 insertions(+), 2 deletions(-)
diff --git a/funasr/utils/load_utils.py b/funasr/utils/load_utils.py
index 52753c4..1d80fcf 100644
--- a/funasr/utils/load_utils.py
+++ b/funasr/utils/load_utils.py
@@ -10,7 +10,6 @@
import time
import logging
from torch.nn.utils.rnn import pad_sequence
-from pydub import AudioSegment
try:
from funasr.download.file import download_from_url
@@ -19,6 +18,11 @@
import pdb
import subprocess
from subprocess import CalledProcessError, run
+
+try:
+ from pydub import AudioSegment
+except:
+ pass
def is_ffmpeg_installed():
@@ -166,7 +170,12 @@
byte_data = BytesIO(input)
# 浣跨敤 pydub 鍔犺浇闊抽
- audio = AudioSegment.from_file(byte_data)
+ try:
+ audio = AudioSegment.from_file(byte_data)
+ except:
+ raise RuntimeError(
+ "You are decoding the pcm data, please install pydub first. via `pip install pydub`."
+ )
# 纭繚閲囨牱鐜囦负 16000 Hz
if audio.frame_rate != fs:
--
Gitblit v1.9.1