LukeGPT88 commited on
Commit
ffbf15d
·
verified ·
1 Parent(s): c169917

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -13
README.md CHANGED
@@ -15,11 +15,8 @@ language:
15
  - en
16
  pipeline_tag: token-classification
17
  ---
18
- # distilbert-NER
19
 
20
- If my open source models have been useful to you, please consider supporting me in building small, useful AI models for everyone (and help me afford med school / help out my parents financially). Thanks!
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("dslim/distilbert-NER")
50
- model = AutoModelForTokenClassification.from_pretrained("dslim/distilbert-NER")
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"