Localized Model Customization

[sulumaenjun]

This project is amazing! How can I customize my own localized model, or download an existing model from Hugging Face to run locally, and how should I configure the model ID? Thanks.

[sabman]

Good question and thanks for the kind words @sulumaenjun Have a quick look at this PR #113 but we'll add a guide for this in the docs probably next week. @mhassanch might have some good ideas too.

[mhassanch]

Here are the steps you need to follow:

• Export your model to ONNX format.
• Create a model directory like this example: geobase/building-detection.
• Place your ONNX model and a preprocessor_config.json file inside this new directory.
• In the model registry, set the model_id to the path of your new model directory.

If you are adding a completely new type of task, you'll also need to:

• Add the new task to src/registry.ts.
• Create a model file in src/models/your_model.ts to handle model initialization, inference, and post-processing.

I hope this helps!