Skip to content

Commit abc5d30

Browse files
committed
fix api torchaudio.load bug
1 parent 01dffe6 commit abc5d30

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

api.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,8 @@
1010
import torchaudio
1111
from model import SenseVoiceSmall
1212
from funasr.utils.postprocess_utils import rich_transcription_postprocess
13+
from io import BytesIO
14+
1315

1416
class Language(str, Enum):
1517
auto = "auto"
@@ -49,9 +51,11 @@ async def turn_audio_to_text(files: Annotated[List[bytes], File(description="wav
4951
audios = []
5052
audio_fs = 0
5153
for file in files:
52-
data_or_path_or_list, audio_fs = torchaudio.load(file)
54+
file_io = BytesIO(file)
55+
data_or_path_or_list, audio_fs = torchaudio.load(file_io)
5356
data_or_path_or_list = data_or_path_or_list.mean(0)
5457
audios.append(data_or_path_or_list)
58+
file_io.close()
5559
if lang == "":
5660
lang = "auto"
5761
if keys == "":

0 commit comments

Comments
 (0)