ulysses115's picture
Update infer.py
844ee1d