XDHDD commited on
Commit
21b67ad
1 Parent(s): 4c87576

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,8 +41,8 @@ def load_model(model):
41
  return session, onnx_model, input_names, output_names
42
 
43
  def inference(re_im, session, onnx_model, input_names, output_names):
44
- inputs = {input_names[i]: torch.from_numpy(np.zeros([d.dim_value for d in _input.type.tensor_type.shape.dim],
45
- dtype=np.float32)).type(torch.float16)
46
  for i, _input in enumerate(onnx_model.graph.input)
47
  }
48
 
 
41
  return session, onnx_model, input_names, output_names
42
 
43
  def inference(re_im, session, onnx_model, input_names, output_names):
44
+ inputs = {input_names[i]: np.zeros([d.dim_value for d in _input.type.tensor_type.shape.dim],
45
+ dtype=np.float32))
46
  for i, _input in enumerate(onnx_model.graph.input)
47
  }
48