Update for latest classify_image code.