oskarastrom's picture
Update inference.py
4b6e487