equ1's picture
Update model.py
e4b9ec4