Fixing code in batch predictions docs
Small change in docs