HuggingFace integration - Githubissues

xavysp / TEED

TEED: Tiny and Efficient Edge Detector

MIT License

178 stars 17 forks source link

HuggingFace integration #23

Open not-lain opened 2 months ago

not-lain commented 2 months ago

This PR will allow your model to be available on huggingface making users to access it with ease. the PyTorchModelHubMixin will add 3 methods to your model similar to the transformers library :

push_to_hub
from_pretrained
save_pretrained

to help you setup your model on the hub you can follow the instructions in this colab notebook : https://colab.research.google.com/drive/1FmInXvJ1v6WJ7TbqXAVvIrth7kwO_Scx?usp=sharing

By using PyTorchModelHub you can get access to :

know how many times your model has been downloaded from huggingface each month
filter huggingface and check how many deet models exist on the hub

If you have any questions or feedback feel free to let me know.

Regards

Hafedh Hichri (Lain)