Fixed predict to work for the first image (problems with second).