kevinwang676's picture
Update infer.py
7c04818