Spaces:
Sleeping
Sleeping
roychao19477
commited on
Commit
·
f9b4de2
1
Parent(s):
d1947d1
Update chunk feature
Browse files
app.py
CHANGED
@@ -108,10 +108,6 @@ def run_avse_inference(video_path, audio_path):
|
|
108 |
# "video_frames": bg_frames[np.newaxis, ...]
|
109 |
#}
|
110 |
|
111 |
-
print(noisy.shape)
|
112 |
-
print(bg_frames.shape)
|
113 |
-
fsefsfefsf
|
114 |
-
|
115 |
with torch.no_grad():
|
116 |
# Version 1
|
117 |
#estimated = avse_model.enhance(data).reshape(-1)
|
@@ -141,6 +137,10 @@ def run_avse_inference(video_path, audio_path):
|
|
141 |
"video_frames": video_chunk[np.newaxis, ...]
|
142 |
}
|
143 |
|
|
|
|
|
|
|
|
|
144 |
with torch.no_grad():
|
145 |
out = avse_model.enhance(data).reshape(-1)
|
146 |
estimated_chunks.append(out)
|
|
|
108 |
# "video_frames": bg_frames[np.newaxis, ...]
|
109 |
#}
|
110 |
|
|
|
|
|
|
|
|
|
111 |
with torch.no_grad():
|
112 |
# Version 1
|
113 |
#estimated = avse_model.enhance(data).reshape(-1)
|
|
|
137 |
"video_frames": video_chunk[np.newaxis, ...]
|
138 |
}
|
139 |
|
140 |
+
print(audio_chunk.shape)
|
141 |
+
print(video_chunk.shape)
|
142 |
+
fsefsef
|
143 |
+
|
144 |
with torch.no_grad():
|
145 |
out = avse_model.enhance(data).reshape(-1)
|
146 |
estimated_chunks.append(out)
|