mgbam's picture
Update inference.py
1e7a57c verified