roychao19477 commited on
Commit
13db68c
·
1 Parent(s): eb529cf

Update module

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(output_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))
 
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))