junaid17's picture
Update inference/predictor.py
3e5dd6d verified