Unable to start the AI server via docker

[aaswan]

Following https://github.com/graykode/ai-docstring#1-run-the-container-for-the-model-inference-server when I run

docker run -it -d -p 5000:5000 --memory 2g --memory-swap -1 graykode/ai-docstring:cpu

I dont see any containers running via

docker ps -a

I got the following logs

docker logs -t <exited container name>

OSError: Can't load config for 'microsoft/codebert-base'. Make sure that: 'microsoft/codebert-base' is a correct model identifier listed on 'https://huggingface.co/models' or 'microsoft/codebert-base' is the correct path to a directory containing a config.json file