Added a new modeling framework in predict.py.