Updates and minor improvements

[mohamad-hasan-sohan-ajini]

Hey

Thanks for your nice repo, although there were some problems in running scripts. This pull request tries to solve some issues regarding:

1- Exact library/famework version used to run scripts. 2- Make model more ONNX compatible. Take a look at issue #35

regards