Imran1's picture
Update code/inference.py
5c36d2f verified