search

DeepLabCut / DLClibrary

DLClibrary is a lightweight library supporting universal functions for the DeepLabCut ecosystem.
GNU Lesser General Public License v3.0
5 stars 4 forks source link

Have Hugging Face Hub Integration #16

Open merveenoyan opened 1 year ago

merveenoyan commented 1 year ago

Hello 👋 As we discussed a week ago, it would be nice to have a hugging face hub integration. This would allow folks to share their DLC models openly and other people can use them as they wish. Folks can filter the models on the bar like they can with e.g. TensorFlow models (see below) There's multiple steps to completely integrate a library but for now, what I mean is to have a push_to_hub and load_from_hub function like this one.

Screenshot 2023-06-26 at 18 04 12

WDYT?

davanstrien commented 1 year ago

This sounds great; just chiming in to say if you want more control of the curation of models in the zoo, you could use an official org for DeepLabCut to host 'canonical' models. This would allow people to share models easily and give you/end users more control over which models have met some quality threshold.

MMathisLab commented 1 year ago

hi @merveenoyan thanks! our integration is here: https://github.com/DeepLabCut/DLClibrary so maybe we can take the convo there? THanks so much! 🤗

AlexEMG commented 1 year ago

Here is how we load models: https://github.com/DeepLabCut/DLClibrary#example-usage

Any suggestions are welcome!