File size: 921 Bytes
6be7cb3 b053c99 6be7cb3 b053c99 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
# 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")
```
|