Gieorgie's picture
Update prediction.py
902db92 verified