ramadhanlmzero commited on
Commit
ff5bd2e
·
1 Parent(s): 85ff7cb
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -14,8 +14,7 @@ ner_model = pipeline(
14
 
15
  topic_model = pipeline(
16
  "text-classification",
17
- model="iqbalpurba26/IndoBERT-topic-classification",
18
- from_tf=True
19
  )
20
 
21
  TOPIC_LABELS = [
 
14
 
15
  topic_model = pipeline(
16
  "text-classification",
17
+ model="w11wo/indonesian-roberta-base-news-topic-classification"
 
18
  )
19
 
20
  TOPIC_LABELS = [