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 |
|