Imran1's picture
Update code/inference.py
73941c5 verified