Update README.md
Browse files
README.md
CHANGED
@@ -15,11 +15,8 @@ language:
|
|
15 |
- en
|
16 |
pipeline_tag: token-classification
|
17 |
---
|
18 |
-
# distilbert-NER
|
19 |
|
20 |
-
|
21 |
-
|
22 |
-
<a href="https://www.buymeacoffee.com/dslim" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/v2/arial-yellow.png" alt="Buy Me A Coffee" style="height: 60px !important;width: 217px !important;" ></a>
|
23 |
|
24 |
## Model description
|
25 |
|
@@ -29,13 +26,6 @@ This model accurately identifies the same four types of entities as its BERT cou
|
|
29 |
|
30 |
The model was fine-tuned on the English version of the [CoNLL-2003 Named Entity Recognition](https://www.aclweb.org/anthology/W03-0419.pdf) dataset, which is widely recognized for its comprehensive and diverse range of entity types.
|
31 |
|
32 |
-
### Available NER models
|
33 |
-
| Model Name | Description | Parameters |
|
34 |
-
|-------------------|-------------|------------------|
|
35 |
-
| [distilbert-NER](https://huggingface.co/dslim/distilbert-NER) | Fine-tuned DistilBERT - a smaller, faster, lighter version of BERT | 66M |
|
36 |
-
| [bert-large-NER](https://huggingface.co/dslim/bert-large-NER/) | Fine-tuned bert-large-cased - larger model with slightly better performance | 340M |
|
37 |
-
| [bert-base-NER](https://huggingface.co/dslim/bert-base-NER)-([uncased](https://huggingface.co/dslim/bert-base-NER-uncased)) | Fine-tuned bert-base, available in both cased and uncased versions | 110M |
|
38 |
-
|
39 |
## Intended uses & limitations
|
40 |
|
41 |
#### How to use
|
@@ -46,8 +36,8 @@ This model can be utilized with the Transformers *pipeline* for NER, similar to
|
|
46 |
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
47 |
from transformers import pipeline
|
48 |
|
49 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
50 |
-
model = AutoModelForTokenClassification.from_pretrained("
|
51 |
|
52 |
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
|
53 |
example = "My name is Wolfgang and I live in Berlin"
|
|
|
15 |
- en
|
16 |
pipeline_tag: token-classification
|
17 |
---
|
|
|
18 |
|
19 |
+
##
|
|
|
|
|
20 |
|
21 |
## Model description
|
22 |
|
|
|
26 |
|
27 |
The model was fine-tuned on the English version of the [CoNLL-2003 Named Entity Recognition](https://www.aclweb.org/anthology/W03-0419.pdf) dataset, which is widely recognized for its comprehensive and diverse range of entity types.
|
28 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
## Intended uses & limitations
|
30 |
|
31 |
#### How to use
|
|
|
36 |
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
37 |
from transformers import pipeline
|
38 |
|
39 |
+
tokenizer = AutoTokenizer.from_pretrained("LukeGPT88/distilbert-NER")
|
40 |
+
model = AutoModelForTokenClassification.from_pretrained("LukeGPT88/distilbert-NER")
|
41 |
|
42 |
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
|
43 |
example = "My name is Wolfgang and I live in Berlin"
|