Update app.py
Browse files
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 |
|