XDHDD commited on
Commit
215b1af
1 Parent(s): f3ee147

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ def inference(re_im, session, onnx_model, input_names, output_names):
50
  output_audio = output_audio.permute(1, 0, 2).contiguous()
51
  output_audio = torch.view_as_complex(output_audio)
52
  output_audio = torch.istft(output_audio, window, stride, window=hann)
53
- return output_audio.np()
54
 
55
  def visualize(hr, lr, recon, sr):
56
  sr = sr
 
50
  output_audio = output_audio.permute(1, 0, 2).contiguous()
51
  output_audio = torch.view_as_complex(output_audio)
52
  output_audio = torch.istft(output_audio, window, stride, window=hann)
53
+ return output_audio.numpy()
54
 
55
  def visualize(hr, lr, recon, sr):
56
  sr = sr