oskarastrom's picture
Update inference.py
2d0ef6d