sleimanitani

**Describe the bug** Inference from a model I trained isn't working and just printing out "Error: 'config'" instead **To Reproduce** I used the colab in the repo, with the change...