Spaces:
Sleeping
Sleeping
roychao19477
commited on
Commit
·
f23d724
1
Parent(s):
58e2fbb
Update module
Browse files
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():
|