roychao19477 commited on
Commit
f23d724
·
1 Parent(s): 58e2fbb

Update module

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -94,8 +94,8 @@ def run_avse_inference(video_path, audio_path):
94
 
95
  # Combine into input dict (match what model.enhance expects)
96
  data = {
97
- "noisy_audio": noisy,
98
- "video_frames": bg_frames
99
  }
100
 
101
  with torch.no_grad():
 
94
 
95
  # Combine into input dict (match what model.enhance expects)
96
  data = {
97
+ "noisy_audio": noisy.numpy(),
98
+ "video_frames": bg_frames.numpy()
99
  }
100
 
101
  with torch.no_grad():