It was a task from the AI Engineer boot camp, if you want to improve this repo, you can contact me, see my code to build this repo via this link :

https://colab.research.google.com/drive/1Nmvnyma8pFovz_WsucC05dJhR5T9DR_I?usp=sharing