Update README.md
Browse files
README.md
CHANGED
|
@@ -55,8 +55,6 @@ def predict_scibert_labels(sentence):
|
|
| 55 |
for i, word_idx in enumerate(word_ids):
|
| 56 |
if word_idx is None or word_idx == previous_word_idx:
|
| 57 |
continue
|
| 58 |
-
token_id = int(inputs['input_ids'][0][i])
|
| 59 |
-
token = tokenizer.convert_ids_to_tokens(token_id)
|
| 60 |
label = id2label[predictions[i]]
|
| 61 |
final_tokens.append(words[word_idx])
|
| 62 |
final_labels.append(label)
|
|
|
|
| 55 |
for i, word_idx in enumerate(word_ids):
|
| 56 |
if word_idx is None or word_idx == previous_word_idx:
|
| 57 |
continue
|
|
|
|
|
|
|
| 58 |
label = id2label[predictions[i]]
|
| 59 |
final_tokens.append(words[word_idx])
|
| 60 |
final_labels.append(label)
|