abc

edge_engine/model_optimizer.py

@@ -36,7 +36,7 @@ class ModelOptimization:
             current_directory = os.getcwd()
 
             logger.info(f"Current directory is :  {current_directory}")
-            build_path = os.path.join(current_directory, "yolov5", "build")
+            build_path = os.path.join("/app/edge_engine", "yolov5", "build")
             if os.path.isdir('/app/edge_engine/yolov5/build'):
                 logger.info("build directory exists. Removing build directory!!")
                 shutil.rmtree('/app/edge_engine/yolov5/build')