Spaces:
Sleeping
Sleeping
update
Browse files
examples/sound_classification_by_lstm/step_9_evaluation_onnx_model.py
CHANGED
|
@@ -73,9 +73,9 @@ def main():
|
|
| 73 |
evaluation_file = tgt_path / "evaluation.xlsx"
|
| 74 |
else:
|
| 75 |
config_file = model_dir / "config.yaml"
|
| 76 |
-
onnx_model_file = args.onnx_model_file
|
| 77 |
-
vocab_path = args.vocabulary_dir
|
| 78 |
-
evaluation_file = args.dataset
|
| 79 |
|
| 80 |
config = WaveClassifierConfig.from_pretrained(config_file.as_posix())
|
| 81 |
ort_session = ort.InferenceSession(onnx_model_file.as_posix())
|
|
|
|
| 73 |
evaluation_file = tgt_path / "evaluation.xlsx"
|
| 74 |
else:
|
| 75 |
config_file = model_dir / "config.yaml"
|
| 76 |
+
onnx_model_file = Path(args.onnx_model_file)
|
| 77 |
+
vocab_path = Path(args.vocabulary_dir)
|
| 78 |
+
evaluation_file = Path(args.dataset)
|
| 79 |
|
| 80 |
config = WaveClassifierConfig.from_pretrained(config_file.as_posix())
|
| 81 |
ort_session = ort.InferenceSession(onnx_model_file.as_posix())
|