Adding support for shared models' url and improving prediction code