gokul commited on
Commit
8d56857
·
verified ·
1 Parent(s): fd1b5cf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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("dslim/distilbert-NER")
77
- model = AutoModelForTokenClassification.from_pretrained("dslim/distilbert-NER")
78
 
79
  nlp = pipeline("ner", model=model, tokenizer=tokenizer)
80
- example = "My name is Wolfgang and I live in Berlin"
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)