infer.py