added file copy logic

yolo_optimization.py

@@ -32,11 +32,12 @@ class ModelOptimization:
     def optimize_model(self):
         try:
 
+
+            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"/home/ilens/yolov5/build/{weight_name_with_extension}"
+            src = f"{current_directory}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")
             os.mkdir(build_path)