Limit number of predicted entities

There's not really an elegant way to do this currently, sorry --- at least not with the main model. It's not a requirement that comes up often.