unionpoint's picture
Update src/infer.py
1454fd3 verified