Update model.py

Changing import statement