equ1's picture
Update model.py
a232c52