quantumbit's picture
Update inference.py
e0f9ffc verified