Maintain the architecture of the original image - mnml.ai