Getting predictions

Configuring custom-trained models