transpolymer's picture
Update prediction.py
6c37c0f verified