XDHDD commited on
Commit
f3ee147
1 Parent(s): 47bb03b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -109,7 +109,7 @@ lossy_input = lossy_input.reshape(-1)
109
  hann = torch.sqrt(torch.hann_window(window))
110
  lossy_input_tensor = torch.tensor(lossy_input)
111
  re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(
112
- 1).np().astype(np.float32)
113
  session, onnx_model, input_names, output_names = load_model()
114
 
115
  if st.button('Сгенерировать потери'):
 
109
  hann = torch.sqrt(torch.hann_window(window))
110
  lossy_input_tensor = torch.tensor(lossy_input)
111
  re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(
112
+ 1).numpy().astype(np.float32)
113
  session, onnx_model, input_names, output_names = load_model()
114
 
115
  if st.button('Сгенерировать потери'):