YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
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 model.
Training Data
Trained on 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")
- Downloads last month
- 3