Mahiruoshi's picture
Update inference.py
3ad53be