oskarastrom's picture
Update inference.py
f0af381