Fixed path to setup.py

scaleoutsystems · Oct 29, 2024 · d43a74d · d43a74d
1 parent 767d5bd
commit d43a74d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/client/setup.py b/client/setup.py
@@ -16,10 +16,10 @@
 test_path = server_config.get("test", "fed_dataset/test/images")
 
 # Paths for model files
-model_folder = "yolov8models"
+model_folder = "client/yolov8models"
 model_file = os.path.join(model_folder, f"{model_size}.yaml")
-output_model_file = "model.yaml"
-output_data_file = "data.yaml"
+output_model_file = "client/model.yaml"
+output_data_file = "client/data.yaml"
 
 # Generate model.yaml with 'nc' on line 4
 if not os.path.exists(model_file):