I run your source code ,but it happen some errors , for example: 1 in mian.py , it would give a error as below:  2 when run predict.py , it would give a error as below: 
I run your source code ,but it happen some errors , for example:

1 in mian.py , it would give a error as below:
2 when run predict.py , it would give a error as below:
