step working on ease of deployment for oak d

config.py

@@ -5,10 +5,10 @@ DEFAULT_IMAGES_DIR = "IMAGE/"  # Directory containing wood defect images
 DEFAULT_MODEL_WEIGHTS = "runs/yolox_training/training/weights/best.pt"  # Trained YOLOX model
 
 # Training defaults
-DEFAULT_TRAIN_EPOCHS = 20
+DEFAULT_TRAIN_EPOCHS = 1
 DEFAULT_BATCH_SIZE = 4
 DEFAULT_LEARNING_RATE = 1e-4
-DEFAULT_MODEL_SIZE = "small"  # nano, small, medium, base
+DEFAULT_MODEL_SIZE = "nano"  # nano, small, medium, base
 
 # Dataset split ratios
 DEFAULT_TRAIN_SPLIT = 0.8