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