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 |
)
|