ml6team
/

keyphrase-extraction-distilbert-kptimes

Token Classification

keyphrase-extraction

Inference Endpoints

Model card Files Files and versions Community

DeDeckerThomas commited on Jun 16, 2022

Commit

b2bdd83

•

1 Parent(s): 12d1f96

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -158,7 +158,7 @@ lbl2idx = {"B": 0, "I": 1, "O": 2}
 idx2label = {0: "B", 1: "I", 2: "O"}
 # Tokenizer
-tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased", add_prefix_space=True)
 max_length = 512
 # Dataset parameters

 idx2label = {0: "B", 1: "I", 2: "O"}
 # Tokenizer
+tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 max_length = 512
 # Dataset parameters