Ani14's picture
Update predict.py
7eb41d6 verified