Spaces:
Runtime error
Runtime error
update model name
Browse files- 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 |
)
|