ML models for image enhancement