guruprasadr commited on
Commit
7c52089
1 Parent(s): 4ababee

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -228,7 +228,7 @@ You can finetune this model on your own dataset.
228
  from span_marker import SpanMarkerModel, Trainer
229
 
230
  # Download from the 🤗 Hub
231
- model = SpanMarkerModel.from_pretrained("YurtsAI/named_entity_recognition_document_context")
232
 
233
  # Specify a Dataset with "tokens" and "ner_tag" columns
234
  dataset = load_dataset("conll2003") # For example CoNLL2003
 
228
  from span_marker import SpanMarkerModel, Trainer
229
 
230
  # Download from the 🤗 Hub
231
+ model = SpanMarkerModel.from_pretrained("YurtsAI/ner-document-context")
232
 
233
  # Specify a Dataset with "tokens" and "ner_tag" columns
234
  dataset = load_dataset("conll2003") # For example CoNLL2003