diff --git a/node/video_split_node.py b/node/video_split_node.py index 33b1236..5b66e02 100644 --- a/node/video_split_node.py +++ b/node/video_split_node.py @@ -1,7 +1,7 @@ import cv2 import torch import numpy as np -import torchaudio + import tempfile import subprocess import os @@ -39,6 +39,7 @@ class SplitVideoByFrames: audio_dict = {"waveform": waveform, "sample_rate": sample_rate} else: try: + import torchaudio waveform, sample_rate = torchaudio.load(audio_path) except Exception as e: print(f"警告:音频文件读取失败,返回空音频。错误信息: {e}") diff --git a/requirements.txt b/requirements.txt index a9b38a6..7e01c2e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,4 +3,6 @@ numpy requests qiniu openai -imageio-ffmpeg \ No newline at end of file +imageio-ffmpeg +torch +torchaudio \ No newline at end of file