transpolymer's picture
Update prediction.py
8e77cfc verified