javiercha commited on
Commit
b0ec70c
1 Parent(s): 3cc479a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ nlp = pipeline('text-classification', model='bdsl/HanmunRoBERTa')
14
 
15
  def predict_century(text):
16
  preprocessed_input = preprocess_text(text)
17
- result = nlp(preprocessed_input)
18
  result.sort(key=lambda x: x['score'], reverse=True)
19
 
20
  scores = {f"{i}th century": 0 for i in range(15, 20)}
 
14
 
15
  def predict_century(text):
16
  preprocessed_input = preprocess_text(text)
17
+ result = nlp(preprocessed_input, top_k=None)
18
  result.sort(key=lambda x: x['score'], reverse=True)
19
 
20
  scores = {f"{i}th century": 0 for i in range(15, 20)}