Update predict_depth.r