From 1da964609d51ffa82ef37ddc471e9c143ec47dbd Mon Sep 17 00:00:00 2001
From: xie junchen <1132348563@qq.com>
Date: Sun, 3 Aug 2025 21:21:14 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 node/video_split_node.py | 3 ++-
 requirements.txt         | 4 +++-
 2 files changed, 5 insertions(+), 2 deletions(-)

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