Update pyinfer.py