Update model.py

fixed a minor error i saw in the uploaded version