Update script.py

script.py

@@ -112,7 +112,7 @@ def train_model(training_config=training_config):
     """
     Train the YOLO model on the given dataset using the provided configuration.
 
-    NOTE: DO NOT MODIFY THIS FUNCTION.
+    NOTE: DO NOT MODIFY THIS FUNCTION AT ALL OR YOUR SCRIPT WILL FAIL.
     """
 
     training_dataset = prepare_dataset()
@@ -134,7 +134,11 @@ def train_model(training_config=training_config):
 
     print("Initializing the YOLO model...")
 
-    model = YOLO("yolov10m.pt")
+    #DO NOT MODIFY THIS LINE
+    model = YOLO(
+        model="/tmp/data/yolo11m.pt",
+        
+    )
     
     print("Model initialized.")