Update README.md
Browse files
README.md
CHANGED
|
@@ -114,7 +114,11 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
| 114 |
model.to(device)
|
| 115 |
model.eval()
|
| 116 |
|
| 117 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 118 |
save_pair = run_ner(texts, idx2label, tokenizer, model, device)
|
| 119 |
|
| 120 |
</code></pre>
|
|
|
|
| 114 |
model.to(device)
|
| 115 |
model.eval()
|
| 116 |
|
| 117 |
+
We recommend to inference by sentences.
|
| 118 |
+
|
| 119 |
+
text = ""
|
| 120 |
+
|
| 121 |
+
texts = text.split('. ')
|
| 122 |
save_pair = run_ner(texts, idx2label, tokenizer, model, device)
|
| 123 |
|
| 124 |
</code></pre>
|