transpolymer's picture
Update prediction.py
cf36af6 verified