Edit model card
YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)

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
39
Inference API
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.