added file copy logic

yolo_optimization.py

@@ -36,7 +36,7 @@ class ModelOptimization:
             current_directory = os.getcwd()
             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/{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")