XDHDD commited on
Commit
e03b423
·
verified ·
1 Parent(s): c2ae7cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -142,6 +142,8 @@ hann = torch.sqrt(torch.hann_window(window))
142
  lossy_input_tensor = torch.tensor(lossy_input)
143
  re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(
144
  1).numpy().astype(np.float32)
 
 
145
 
146
  session, onnx_model, input_names, output_names = load_model(model_ver)
147
 
 
142
  lossy_input_tensor = torch.tensor(lossy_input)
143
  re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(
144
  1).numpy().astype(np.float32)
145
+ re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(
146
+ 1).numpy().astype(np.float16)
147
 
148
  session, onnx_model, input_names, output_names = load_model(model_ver)
149