added file copy logic

yolo_optimization.py

@@ -34,13 +34,14 @@ class ModelOptimization:
 
 
             current_directory = os.getcwd()
+
+            logger.info(f"Current directory is :  {current_directory}")
+            build_path = os.path.join(current_directory, "yolov5", "build")
+            os.mkdir(build_path)
             weight_name_with_extension = os.path.basename(self.weight_path)
             weight_name, extension = os.path.splitext(weight_name_with_extension)
             src = f"{current_directory}/yolov5/build/"
             shutil.copy(self.weight_path, src)
-            logger.info(f"Current directory is :  {current_directory}")
-            build_path = os.path.join(current_directory, "yolov5", "build")
-            os.mkdir(build_path)
             logger.info(f"Created build folder")
             os.chdir('yolov5/build')
             logger.info("Running CMake command")