The solution is provided in a Jupyter Notebook. To run the code locally follow the steps below.

pip install -r requirements.txt jupyter lab