ntt123's picture
Update inference.py
f1cf79d