cindyangelira commited on
Commit
c6d00dd
·
verified ·
1 Parent(s): 71fe727

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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():