updates

script.py

@@ -61,10 +61,19 @@ def export_to_yolo_format(
             split=split
         )
 
-def train_model(training_dataset, training_config):
+def train_model():
     """
     Train the YOLO model on the given dataset using the provided configuration.
     """
+
+    script_dir = os.path.dirname(os.path.abspath(__file__))
+
+    config_path = os.path.join(script_dir, 'training_config.yaml')
+
+    with open(config_path, 'r') as file:
+        training_config = yaml.safe_load(file)
+
+
     training_dataset = fouh.load_from_hub(
         "Voxel51/Data-Centric-Visual-AI-Challenge-Train-Set",
         max_samples=100 #for testing remove this later