From b5eb4179624da21250aecf0e3b24f045c5fcd5c0 Mon Sep 17 00:00:00 2001 From: cbpark-nota Date: Wed, 28 Aug 2024 09:02:07 +0000 Subject: [PATCH] Update readme and single execution --- README.md | 5 ++--- netspresso_inference_package/inference/inference_service.py | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index c9cba17..12b0127 100644 --- a/README.md +++ b/README.md @@ -13,9 +13,8 @@ python3 inference.py --model_file_path tests/your_model_file.tflite --dataset_fi ```python from netspresso_inference_package.inference.inference_service import InferenceService inf_service = InferenceService( - model_file_path="/app/tests/people_detection.onnx", - dataset_file_path="/app/tests/dataset_for_onnx.npy" + model_file_path="/app/tests/people_detection.onnx" ) -inf_service.run() +inf_service.run(dataset_file_path="/app/tests/dataset_for_onnx.npy") print(inf_service.result_file_path) ``` \ No newline at end of file diff --git a/netspresso_inference_package/inference/inference_service.py b/netspresso_inference_package/inference/inference_service.py index 92f1ce2..26b0b67 100644 --- a/netspresso_inference_package/inference/inference_service.py +++ b/netspresso_inference_package/inference/inference_service.py @@ -68,7 +68,6 @@ def run(self, dataset_file_path): if __name__ == "__main__": inf_service = InferenceService( - model_file_path="/app/tests/people_detection.onnx", - dataset_file_path="/app/tests/dataset_for_onnx.npy" + model_file_path="/app/tests/people_detection.onnx" ) - inf_service.run() \ No newline at end of file + inf_service.run(dataset_file_path="/app/tests/dataset_for_onnx.npy") \ No newline at end of file