Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ def load_model(model):
|
|
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 |
|
@@ -116,7 +116,7 @@ st.audio(uploaded_file)
|
|
116 |
|
117 |
model_ver = st.selectbox(
|
118 |
'Оригинал или Pruned ?',
|
119 |
-
('frn.onnx', 'frn_modified.onnx', 'frn_out_Q.onnx', 'frn_out_opset_v.onnx', 'frn_fp16.onnx'))
|
120 |
|
121 |
st.write('Вы выбрали:', model_ver)
|
122 |
|
|
|
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 |
|
|
|
116 |
|
117 |
model_ver = st.selectbox(
|
118 |
'Оригинал или Pruned ?',
|
119 |
+
('frn.onnx', 'frn_modified.onnx', 'frn_out_Q.onnx', 'frn_out_opset_v.onnx', 'frn_fp16 (1).onnx'))
|
120 |
|
121 |
st.write('Вы выбрали:', model_ver)
|
122 |
|