slkML's picture
updated model.py
aa11a0b verified