-
Notifications
You must be signed in to change notification settings - Fork 88
Open
Description
We should add a huggingface integration to publish micro-sam and related models. See also computational-cell-analytics/medico-sam#86.
I have started with the first steps for it by:
- Creating a huggingface organization: https://huggingface.co/computational-cell-analytics
- Creating a repository for medico-sam: https://huggingface.co/computational-cell-analytics/medico-sam
- A branch for working on the integration: https://github.com/computational-cell-analytics/micro-sam/tree/hf
For the next steps we need to decide how to best represent the model in HuggingFace. This is a bit more complex, because our models consist of the SAM model (prompt encoder, image encoder, mask decoder) + a custom UNETR decoder. So I am not sure if we can use the PyTorchModelHubMixin or should rather opt for a custom upload solution. See https://huggingface.co/docs/hub/models-uploading for details.
Once we have figured this out:
- Implement code for the integration so that we can easily upload different models.
- Upload the MedicoSAM model.
- Implement a demo for https://huggingface.co/spaces
- Upload other models and associated demos.
anwai98
Metadata
Metadata
Assignees
Labels
No labels