Adapt to latest Library

[Fliw]

Thank you for creating this deep learning model. I wanted to try it on my local machine, but some of the dependencies were outdated and conflicted with each other. Therefore, I made a simple update to the library, upgrading each dependency to the latest version and adjusting the parameters and values from the old ones to the new ones. I have tested it, and it runs smoothly on my machine.