Ani14's picture
Update predict.py
3ff0ea1 verified