oskarastrom's picture
Update inference.py
0d4fbd4