Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
|
|
3 |
from spacy import displacy
|
4 |
|
5 |
# load model pipeline globally
|
6 |
-
ner_pipe = pipeline("token-classification", model="cindyangelira/ner-roberta-large-bahasa-indonesia", aggregation_strategy = "average")
|
7 |
|
8 |
# define colors for each tag
|
9 |
def get_colors():
|
|
|
3 |
from spacy import displacy
|
4 |
|
5 |
# load model pipeline globally
|
6 |
+
ner_pipe = pipeline("token-classification", model="cindyangelira/ner-roberta-large-bahasa-indonesia-finetuned", aggregation_strategy = "average")
|
7 |
|
8 |
# define colors for each tag
|
9 |
def get_colors():
|