How to use ml6team/keyphrase-extraction-distilbert-openkp with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="ml6team/keyphrase-extraction-distilbert-openkp")
# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("ml6team/keyphrase-extraction-distilbert-openkp") model = AutoModelForTokenClassification.from_pretrained("ml6team/keyphrase-extraction-distilbert-openkp", device_map="auto")
