gokul commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -73,11 +73,11 @@ Use the code below to get started with the model.
|
|
| 73 |
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
| 74 |
from transformers import pipeline
|
| 75 |
|
| 76 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 77 |
-
model = AutoModelForTokenClassification.from_pretrained("
|
| 78 |
|
| 79 |
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
|
| 80 |
-
example = "
|
| 81 |
|
| 82 |
ner_results = nlp(example)
|
| 83 |
print(ner_results)
|
|
|
|
| 73 |
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
| 74 |
from transformers import pipeline
|
| 75 |
|
| 76 |
+
tokenizer = AutoTokenizer.from_pretrained("gokulsarvesh/bert_base_uncase_Conll2012")
|
| 77 |
+
model = AutoModelForTokenClassification.from_pretrained("gokulsarvesh/bert_base_uncase_Conll2012")
|
| 78 |
|
| 79 |
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
|
| 80 |
+
example = "Plan a trip from W Broad Street Columbus to Hazel Street Louisville Kentucky USA on Sunday at 5 o Clock"
|
| 81 |
|
| 82 |
ner_results = nlp(example)
|
| 83 |
print(ner_results)
|