XDHDD commited on
Commit
c2ae7cd
·
verified ·
1 Parent(s): 14fdc6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -45,7 +45,10 @@ def inference(re_im, session, onnx_model, input_names, output_names):
45
  dtype=np.float32)
46
  for i, _input in enumerate(onnx_model.graph.input)
47
  }
48
-
 
 
 
49
  output_audio = []
50
  for t in range(re_im.shape[0]):
51
  inputs[input_names[0]] = re_im[t]
 
45
  dtype=np.float32)
46
  for i, _input in enumerate(onnx_model.graph.input)
47
  }
48
+ inputs = {input_names[i]: np.zeros([d.dim_value for d in _input.type.tensor_type.shape.dim],
49
+ dtype=np.float16)
50
+ for i, _input in enumerate(onnx_model.graph.input)
51
+ }
52
  output_audio = []
53
  for t in range(re_im.shape[0]):
54
  inputs[input_names[0]] = re_im[t]