Imran1's picture
Update inference.py
edaf8b6 verified