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