Properly update the prediction info.

This lets us predict more than one image.