Ani14's picture
Update predict.py
dd6e4e2 verified