Update app.py

app.py

@@ -49,6 +49,7 @@ def inference(re_im, session, onnx_model, input_names, output_names):
     output_audio = []
     for t in range(re_im.shape[0]):
         inputs[input_names[0]] = re_im[t]
+        inputs = torch.from_numpy(inputs, dtype=torch.float16)
         out, prev_mag, predictor_state, mlp_state = session.run(output_names, inputs)
         inputs[input_names[1]] = prev_mag
         inputs[input_names[2]] = predictor_state