From 23e7ddebccd3b05cf7ef89809bcfe565ad6dfa1f Mon Sep 17 00:00:00 2001
From: majic31 <majic31@163.com>
Date: 星期二, 24 十二月 2024 10:00:14 +0800
Subject: [PATCH] Fix the variable name (#2328)
---
funasr/utils/load_utils.py | 23 ++++++++++++++++++-----
1 files changed, 18 insertions(+), 5 deletions(-)
diff --git a/funasr/utils/load_utils.py b/funasr/utils/load_utils.py
index aabf5d9..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():
@@ -138,7 +142,10 @@
def load_bytes(input):
- input = validate_frame_rate(input)
+ try:
+ input = validate_frame_rate(input)
+ except:
+ pass
middle_data = np.frombuffer(input, dtype=np.int16)
middle_data = np.asarray(middle_data)
if middle_data.dtype.kind not in "iu":
@@ -153,16 +160,22 @@
array = np.frombuffer((middle_data.astype(dtype) - offset) / abs_max, dtype=np.float32)
return array
+
def validate_frame_rate(
input,
fs: int = 16000,
):
-
+
# 灏嗘枃浠惰鍙栦负瀛楄妭娴�
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:
@@ -175,7 +188,7 @@
# 鑾峰彇閲嶆柊閲囨牱鍚庣殑瀛楄妭娴佹暟鎹�
input = output.read()
-
+
return input
--
Gitblit v1.9.1