transpolymer's picture
Update prediction.py
4b283df verified