python infer.py --model "base"