ntt123's picture
Update inference.py
a98982f