Update predict_new.py