diff --git a/train.py b/train.py
new file mode 100644
index 0000000..bd4d2e0
--- /dev/null
+++ b/train.py
@@ -0,0 +1,6 @@
+from ultralytics import YOLO
+
+# Load a model
+model = YOLO('yolov8.yaml')
+
+results = model.train(data='data/data.yaml', epochs=10, imgsz=640, device='mps', workers=0)