Update app.py

app.py

@@ -137,10 +137,7 @@ lossy_input *= mask
 lossy_input = lossy_input.reshape(-1)
 hann = torch.sqrt(torch.hann_window(window))
 lossy_input_tensor = torch.tensor(lossy_input)
-#re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(
-    #1).numpy().astype(np.float32)
-re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(
-    1).numpy().astype(torch.float16)
+re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(1).numpy().astype(np.float32)
 
 session, onnx_model, input_names, output_names = load_model(model_ver)