transpolymer's picture
Update prediction.py
c615f5a verified