| # 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") | |
| ``` | |