roychao19477 commited on
Commit
cd0a759
·
1 Parent(s): dba6227

Upload to debug

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -136,7 +136,7 @@ def run_avse_inference(video_path, audio_path):
136
  est = avse_model.enhance(chunk_data).reshape(-1)
137
  estimated_chunks.append(est)
138
 
139
- estimated = torch.cat(estimated_chunks).cpu().numpy()
140
 
141
  # Save result
142
  tmp_wav = audio_path.replace(".wav", "_enhanced.wav")
 
136
  est = avse_model.enhance(chunk_data).reshape(-1)
137
  estimated_chunks.append(est)
138
 
139
+ estimated = np.concatenate(estimated_chunks, axis=0)
140
 
141
  # Save result
142
  tmp_wav = audio_path.replace(".wav", "_enhanced.wav")