ml6team
/

keyphrase-extraction-distilbert-inspec

Token Classification

keyphrase-extraction

Inference Endpoints

Model card Files Files and versions Community

DeDeckerThomas commited on Jun 16, 2022

Commit

d7a71e2

•

1 Parent(s): a06a57f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -162,7 +162,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