Updated prediction API sample to v1.6