tiny modification to imgnet_prediction