transpolymer's picture
Update prediction.py
dd5919f verified