XDHDD commited on
Commit
3ebd123
1 Parent(s): b3d7ba3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +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 = output_audio.detach().numpy()
65
  output_audio = torch.istft(output_audio, window, stride, window=hann)
66
  return output_audio.numpy()
67
 
 
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.to(torch.float32)
65
  output_audio = torch.istft(output_audio, window, stride, window=hann)
66
  return output_audio.numpy()
67