You can indeed use your own models! I’ve added instructions on how to do this here: https://gist.github.com/jasonbosco/f4187f6b4f585d2dc8902af85408994a#using-your-own-custom-models