Update module

app.py

@@ -100,7 +100,8 @@ def run_avse_inference(video_path, audio_path):
     }
 
     with torch.no_grad():
-        estimated = avse_model.enhance(data).reshape(-1).cpu().numpy()
+        estimated = avse_model.enhance(data).reshape(-1)
+        print(estimated.shape)
 
     # Save result
     tmp_wav = audio_path.replace(".wav", "_enhanced.wav")