equ1's picture
Update model.py
9b30bbe