transpolymer's picture
Update prediction.py
26505ef verified