transpolymer's picture
Update prediction.py
eea9e94 verified