Spaces:
Sleeping
Sleeping
roychao19477
commited on
Commit
·
13db68c
1
Parent(s):
eb529cf
Update module
Browse files
app.py
CHANGED
@@ -88,7 +88,7 @@ def run_avse_inference(video_path, audio_path):
|
|
88 |
noisy = wavfile.read(audio_path)[1].astype(np.float32) / (2 ** 15)
|
89 |
|
90 |
# Load grayscale video
|
91 |
-
vr = VideoReader(
|
92 |
frames = vr.get_batch(list(range(len(vr)))).asnumpy()
|
93 |
bg_frames = np.array([
|
94 |
cv2.cvtColor(frames[i], cv2.COLOR_RGB2GRAY) for i in range(len(frames))
|
|
|
88 |
noisy = wavfile.read(audio_path)[1].astype(np.float32) / (2 ** 15)
|
89 |
|
90 |
# Load grayscale video
|
91 |
+
vr = VideoReader(video_path, ctx=cpu(0))
|
92 |
frames = vr.get_batch(list(range(len(vr)))).asnumpy()
|
93 |
bg_frames = np.array([
|
94 |
cv2.cvtColor(frames[i], cv2.COLOR_RGB2GRAY) for i in range(len(frames))
|