oskarastrom's picture
Update inference.py
36bc9cc