Update readme and single execution

nota-github · Aug 28, 2024 · b5eb417 · b5eb417
1 parent 370972f
commit b5eb417
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git 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)
 ```
diff --git 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()
+    inf_service.run(dataset_file_path="/app/tests/dataset_for_onnx.npy")