oskarastrom's picture
Update inference.py
1a4c886