Update predict_depth.sh