Imran1's picture
Update code/inference.py
2744dc1 verified