aimlnerd commited on
Commit
c09f7da
1 Parent(s): 02c62ed

update model name

Browse files
Files changed (1) hide show
  1. gradio_ner.py +1 -1
gradio_ner.py CHANGED
@@ -73,7 +73,7 @@ Notice of Representation
73
  John D Locke, Esq"""
74
  ]
75
  # Replace this with your own checkpoint
76
- model_checkpoint = "aimlnerd/bert-finetuned-legalentity-ner"
77
  ner_pipeline = pipeline(
78
  "token-classification", model=model_checkpoint, aggregation_strategy="simple"
79
  )
 
73
  John D Locke, Esq"""
74
  ]
75
  # Replace this with your own checkpoint
76
+ model_checkpoint = "aimlnerd/bert-finetuned-legalentity-ner-accelerate"
77
  ner_pipeline = pipeline(
78
  "token-classification", model=model_checkpoint, aggregation_strategy="simple"
79
  )