Update pred_model_source.py