XDHDD commited on
Commit
b31a744
1 Parent(s): b8d1631

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -137,10 +137,7 @@ lossy_input *= mask
137
  lossy_input = lossy_input.reshape(-1)
138
  hann = torch.sqrt(torch.hann_window(window))
139
  lossy_input_tensor = torch.tensor(lossy_input)
140
- #re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(
141
- #1).numpy().astype(np.float32)
142
- re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(
143
- 1).numpy().astype(torch.float16)
144
 
145
  session, onnx_model, input_names, output_names = load_model(model_ver)
146
 
 
137
  lossy_input = lossy_input.reshape(-1)
138
  hann = torch.sqrt(torch.hann_window(window))
139
  lossy_input_tensor = torch.tensor(lossy_input)
140
+ re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(1).numpy().astype(np.float32)
 
 
 
141
 
142
  session, onnx_model, input_names, output_names = load_model(model_ver)
143