smjain's picture
Update infer.py
032103e verified