transpolymer's picture
Update prediction.py
922a3c0 verified