search

md-mohaiminul / VideoRecap

MIT License
154 stars 6 forks source link

Model repositories on Hugging Face #2

Open merveenoyan opened 6 months ago

merveenoyan commented 6 months ago

Hello 👋 It would be great to have the weights on Hugging Face Hub with a model card that has a note on how to use the model. Drive or GitHub aren't the best place to share weights, so I was wondering if you'd be down. Moreover, you could host a demo built with Gradio on Hugging Face Spaces as well (I can help you out if you need it!) Would you be interested?

md-mohaiminul commented 5 months ago

Hi, Thanks for reaching out. Both sounds great! Please let me know what I should do to move the model on Hugging Face Hub with a model card. Also, if you could help me with the Gradio by sending any pointers, that would be great. Thanks!

merveenoyan commented 5 months ago

Hiya, thanks a lot for the response!

You can extend your main model class like this. This is a mixin to load the model from a repository (in the same PR you can see how to load a model once it's in a repository as well). The repository will be under your namespace.

The repository can be created and files can be uploaded as shown here. This might seem a lot but overall this would take 5 mins.

Easiest way to create a Space is through Gradio Inferface. You can create an Interface as shown here. Once you build the interface, you can host it as shown here.

A bit of a not so relevant note: I saw you are having all the other models in your repository like OpenAI CLIP. This model exists under transformers as well, and I think it would greatly simplify your codebase and make your life easier to use it from transformers like in here.