Hi, thanks for providing this project. When running the inference.py script I noticed that the seaborn dependency could not be found after installing via conda. I see in the Dockerfile that seaborn is explicitly installed after creating the environment, is there any reason why this shouldn't be included in the main environment.yml list of dependencies? I'm happy to provide a PR if so.