Enhancing ML model efficiency with images