from ultralytics import YOLO

# โหลดโมเดล YOLOv8 เปล่า
model = YOLO("yolov8n.pt")  # หรือ yolov8s.pt ถ้าเครื่องแรงขึ้น

# เทรนโมเดลใหม่ด้วย dataset ที่โหลดมา
model.train(
    data="leukemia-detection/data.yaml",  # path ไปยัง dataset.yaml
    epochs=50,
    imgsz=640,
    batch=8,
    name="leukemia-yolo-model",
)