Spaces:
Sleeping
Sleeping
roychao19477
commited on
Commit
·
6e97a1b
1
Parent(s):
f9b4de2
Update chunk feature
Browse files
app.py
CHANGED
@@ -112,7 +112,7 @@ def run_avse_inference(video_path, audio_path):
|
|
112 |
# Version 1
|
113 |
#estimated = avse_model.enhance(data).reshape(-1)
|
114 |
# Version 2
|
115 |
-
chunk_sec =
|
116 |
sr = 16000
|
117 |
audio_chunk_len = chunk_sec * sr # 48000
|
118 |
video_chunk_len = chunk_sec * 25 # 75
|
@@ -137,10 +137,6 @@ def run_avse_inference(video_path, audio_path):
|
|
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)
|
|
|
112 |
# Version 1
|
113 |
#estimated = avse_model.enhance(data).reshape(-1)
|
114 |
# Version 2
|
115 |
+
chunk_sec = 6
|
116 |
sr = 16000
|
117 |
audio_chunk_len = chunk_sec * sr # 48000
|
118 |
video_chunk_len = chunk_sec * 25 # 75
|
|
|
137 |
"video_frames": video_chunk[np.newaxis, ...]
|
138 |
}
|
139 |
|
|
|
|
|
|
|
|
|
140 |
with torch.no_grad():
|
141 |
out = avse_model.enhance(data).reshape(-1)
|
142 |
estimated_chunks.append(out)
|