GPT-2-Simple not being downloaded during deploy

Hello!

I trained a GPT-2 model using gpt-2-simple and have been able to get a Streamlit app running locally that uses the model to generate text.

However, when I deploy it using Streamlit, I get the below error saying the gpt_2_simple module isn’t found.

As you can see in my environment.yml file, I’ve included gpt-2-simple as a dependency and the deployment logs show it’s being downloaded.

What am I missing?

Thanks,
Jordan