transpolymer's picture
Update prediction.py
af4f3b0 verified