Update app.py

app.py

@@ -49,7 +49,8 @@ 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)
+        inputs = torch.from_numpy(inputs)
+        inputs = inputs.type(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