smjain's picture
Update infer.py
9db8a10 verified