Add custom model uploads

For users who train their own models, it would be nice to be able to upload your own checkpoints and use them here.