api changes

yolo_optimization.py

@@ -46,7 +46,7 @@ class ModelOptimization:
             shutil.copy(self.weight_path, os.path.join(code_directory, 'tensorrtx/yolov5/build'))
             logger.info("Copied wts file to build directory")
             weight_name_with_extension = os.path.basename(self.weight_path)
-            os.chdir('tensorrtx/yolov5/build')
+            os.chdir(os.path.join(code_directory, 'tensorrtx/yolov5/build'))
             logger.info("Changed to build directory")
             logger.info("Running CMake command")
             subprocess.run(['cmake', '..'])