XzJosh's picture
Update infer.py
d16354d