added file copy logic

yolo_optimization.py

@@ -31,9 +31,11 @@ class ModelOptimization:
 
     def optimize_model(self):
         try:
-            shutil.copy(self.weight_path, 'yolov5/build')
+
             weight_name_with_extension = os.path.basename(self.weight_path)
             weight_name, extension = os.path.splitext(weight_name_with_extension)
+            src = f"yolov5/build/{weight_name_with_extension}"
+            shutil.copy(self.weight_path, src)
             current_directory = os.getcwd()
             logger.info(f"Current directory is :  {current_directory}")
             build_path = os.path.join(current_directory, "yolov5", "build")