oskarastrom's picture
Update inference.py
fa2b33e