Imran1's picture
Update code/inference.py
6c47fa9 verified