oskarastrom's picture
Update inference.py
a2da5d3