transpolymer's picture
Update prediction.py
a715cd6 verified