import other models using python interface

Hi, similar to https://github.com/brycedrennan/imaginAIry/issues/328 but for the python API .

How can I load a custom model? in particular, I don't know how to pass the model-arch thanks!