ntt123's picture
Update inference.py
b989f92