ntt123's picture
Update inference.py
ddf7e7d