updates

script.py

@@ -74,11 +74,7 @@ def train_model(training_dataset, training_config):
         classes=training_dataset.default_classes,
     )
 
-    model = YOLO("yolov8m.pt")
-
-    # Check if epochs in train_params exceeds 50
-    if 'epochs' in training_config['train_params'] and training_config['train_params']['epochs'] > 50:
-        raise ValueError("Number of epochs cannot exceed 50. Please adjust the 'epochs' parameter in your training configuration.")
+    model = YOLO("yolov10m.pt")
 
     results = model.train(
         data="./yolo_formatted/dataset.yaml",