Imran1's picture
Update code/inference.py
f74b603 verified