roychao19477 commited on
Commit
f9b4de2
·
1 Parent(s): d1947d1

Update chunk feature

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