transpolymer's picture
Update prediction.py
23e592f verified