transpolymer's picture
Update prediction.py
a08c2a4 verified