code reviewed

2025-07-22 12:28:57 +01:00
1 changed files with 5 additions and 0 deletions
@@ -36,8 +36,13 @@ def train_model():
        cache='disk',        # Changed to disk caching for deterministic results
    )
     # Create model directory if it doesn't exist
    import os
    os.makedirs('model/weights', exist_ok=True)
    # Save the trained model
    model.save('model/weights/best.pt')
 if __name__ == '__main__':
    train_model()