from ultralytics import YOLO
import os

# Define the correct path to config.yaml (in the root directory)
config_path = '/config.yaml'

# Load YOLO model
model = YOLO("yolo11n.yaml")  # You can choose a different model type like yolo5n, yolo6n, etc.

# Train the model
results = model.train(data=config_path, epochs=1)