transpolymer's picture
Update prediction.py
e05cbe5 verified