Pytorch-Medical-Segmentation icon indicating copy to clipboard operation
Pytorch-Medical-Segmentation copied to clipboard

Sharing models through Hugging Face Hub

Open omarespejel opened this issue 2 years ago • 1 comments

Hi MontaEllis team!

The Medical Segmentation project is fantastic. I see you host and share models with your server. Would you be interested in sharing your models in the Hugging Face Hub?

This integration would allow you to freely download/upload models and make your work more accessible and visible to the rest of the ML community. We can help you set up a MontaEllis organization (examples, Facebook AI y Stanford NLP).

Creating the repos and adding new models should be a relatively straightforward process. This is a step-by-step guide explaining the process in case you're interested. Please let us know if you would be interested and if you have any questions.

Some of the benefits of sharing your models through the Hub would be:

  • Presence in the HF Hub might lower the entry of barrier to your medical models and increase their visibility.
    • Repos provide valuable metadata about their tasks, languages, metrics, etc. that make them discoverable
  • versioning, commit history, and diffs.
  • multiple features from TensorBoard visualizations, PapersWithCode integration, and more.

Additionally, we have a library to access repositories programmatically (both downloading pre-trained models and pushing, with many nice things such as filtering, caching, etc.). If we want to try out this integration, I would suggest you add one or two models manually and then use the huggingface_hub library to implement downloading those models programmatically from pytorch-medical-segmentation. To read more about it, you might want to check our documentation to read more about it.

Relevant references:

Happy to hear your thoughts,

Omar and the Hugging Face team (cc @osanseviero @abidlabs )

omarespejel avatar Mar 23 '22 01:03 omarespejel

Hi @MontaEllis!

We are very interested in giving even more visibility and support to the deep learning for the healthcare community and your segmentation models are key .

We have new collaboration features in the Hub: pull requests and discussions (link to announcement).

This would allow medical segmentation users to submit, edit, or discuss models within a possible MontaEllis organization in the Hub.

Also, if you want, we can allow image segmentation inference directly from the browser, for example with DETR, for MontaEllis models.

Please let me know if we can help further. Kind regards, Omar

omarespejel avatar May 27 '22 02:05 omarespejel