python predict.py --model_dir=../../output