transpolymer's picture
Update prediction.py
bc443f4 verified