XDHDD commited on
Commit
b3d7ba3
1 Parent(s): 37ff07b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -61,6 +61,7 @@ def inference(re_im, session, onnx_model, input_names, output_names):
61
  output_audio = torch.tensor(np.concatenate(output_audio, 0))
62
  output_audio = output_audio.permute(1, 0, 2).contiguous()
63
  output_audio = torch.view_as_complex(output_audio)
 
64
  output_audio = torch.istft(output_audio, window, stride, window=hann)
65
  return output_audio.numpy()
66
 
 
61
  output_audio = torch.tensor(np.concatenate(output_audio, 0))
62
  output_audio = output_audio.permute(1, 0, 2).contiguous()
63
  output_audio = torch.view_as_complex(output_audio)
64
+ output_audio = output_audio.detach().numpy()
65
  output_audio = torch.istft(output_audio, window, stride, window=hann)
66
  return output_audio.numpy()
67