models: add support for Hugging Face model hub #1974
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Hi,
this PR allows to use trained sequence tagging from the Hugging Face model hub 🤗
It is only needed to specify the namespace, like
stefan-it/flair-ner-conll03
and a model will be downloaded (incl. caching) that can be used in Flair!Naming convention is, that the model file on the model hub is named
model.bin
.The following code snippet shows how to use this new feature:
It is also possible to pass a specific commit or branch name via
@
-syntax:stefan-it/flair-ner-conll03@main
.On the Hugging Face model hub itself, it is also possible to add a
README
file, that gives more details about a trained model:https://huggingface.co/stefan-it/flair-ner-conll03
Many thanks to the Hugging Face team for the awesome model hub ❤️