| # Model Card for Model ID |
|
|
| This model is meant to extract keywords from text. |
|
|
| - **Model type:** text-classification |
| - **Language(s) (NLP):** English |
| - **License:** cc |
| - **Finetuned from model [optional]:** [More Information Needed] |
|
|
| ## Training Details |
|
|
| This model is a fine-tuned version of the [distilbert-base-uncased-finetuned-sst-2-english](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english) model. |
|
|
| ## Training Data |
|
|
| Trained on [51la5/keyword-extraction](https://huggingface.co/datasets/51la5/keyword-extraction) from HuggingFace Hub. |
|
|
| ## How to Get Started with the Model |
|
|
| Note: model inputs were tokenized using distilbert-base-uncased tokenizer |
|
|
| ``` |
| from transformers import AutoTokenizer, AutoModelForSequenceClassification |
| |
| tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased") |
| model = AutoModelForSequenceClassification.from_pretrained("mayapapaya/Keyword-Extractor") |
| ``` |
|
|
